From fe36d1261dc96004e4d4e692a65e1bc53137726b Mon Sep 17 00:00:00 2001
From: Eddy Pedroni <epedroni@pm.me>
Date: Tue, 24 Sep 2024 17:57:01 +0200
Subject: Implement brutal scheduler

---
 src/scheduler_brutal_unittest.py | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)

(limited to 'src/scheduler_brutal_unittest.py')

diff --git a/src/scheduler_brutal_unittest.py b/src/scheduler_brutal_unittest.py
index adfb417..779d2c3 100644
--- a/src/scheduler_brutal_unittest.py
+++ b/src/scheduler_brutal_unittest.py
@@ -6,6 +6,32 @@ from card import Card
 # Force HISTORY_DEPTH to simplify testing
 scheduler_brutal.HISTORY_DEPTH = 3
 
+#--------------------------------------------------------------------------
+# Scheduling behaviour
+#--------------------------------------------------------------------------
+def test_scheduling():
+    cards = {id: Card("", "") for id in range(0, 10)}
+    state = {
+        0: [1, 1, 1],
+        1: [0, 0, 0],
+        2: [0, 0, 1],
+        3: [1, 0, 0],
+
+        4: [None, None, 1   ],
+        5: [None, 1,    None],
+        6: [1,    None, None],
+        7: [None, None, 0   ],
+        8: [0,    0,    None],
+        9: [None, None, None],
+    }
+
+    expected_priority = [9, 6, 5, 7, 8, 4, 1, 3, 2, 0]
+
+    uut = UUT(cards, state)
+
+    for i in range(0, len(expected_priority)):
+        assert set(uut.practice(i + 1)) == set(expected_priority[0:i + 1])
+
 #--------------------------------------------------------------------------
 # State update
 #--------------------------------------------------------------------------
-- 
cgit v1.2.3