update evilnkode.ipynb tests

src/keypad/tower_shuffle.py

@@ -1,6 +1,7 @@
 from dataclasses import dataclass
 import numpy as np
 
+
 @dataclass
 class Tower:
     floors: list[np.ndarray]
@@ -28,7 +29,6 @@ class Tower:
         balance[odd_floors] = 1
         return balance.tolist()
 
-
     def update_tower(self, keep: np.ndarray, other_discard: np.ndarray):
         new_floors = []
         for floor in self.floors:
@@ -51,8 +51,10 @@ class Tower:
             tower.extend(floor.tolist())
         return tower
 
+
 @dataclass
 class TowerShuffle:
+    # TODO: I don't think total_positions is used anywhere
     total_positions: int
     left_tower: Tower
     right_tower: Tower

tests/test_tower_shuffle.py

@@ -1,7 +1,8 @@
 from src.keypad.tower_shuffle import TowerShuffle
 
+
 def test_tower_shuffle():
-    tower = TowerShuffle.new(9)
+    tower = TowerShuffle.new(13)
     print(tower)
     for _ in range(100):
         tower.shuffle()