implement cli for benchmarking and evilnkode

src/keypad/tower_shuffle.py

@@ -1,7 +1,6 @@
 from dataclasses import dataclass
 import numpy as np
 
-
 @dataclass
 class Tower:
     floors: list[np.ndarray]
@@ -29,6 +28,7 @@ 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,10 +51,8 @@ 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,8 +1,7 @@
 from src.keypad.tower_shuffle import TowerShuffle
 
-
 def test_tower_shuffle():
-    tower = TowerShuffle.new(13)
+    tower = TowerShuffle.new(9)
     print(tower)
     for _ in range(100):
         tower.shuffle()