diff options
Diffstat (limited to 'src/scheduler_brutal_unittest.py')
| -rw-r--r-- | src/scheduler_brutal_unittest.py | 52 |
1 files changed, 26 insertions, 26 deletions
diff --git a/src/scheduler_brutal_unittest.py b/src/scheduler_brutal_unittest.py index 779d2c3..8dc72fe 100644 --- a/src/scheduler_brutal_unittest.py +++ b/src/scheduler_brutal_unittest.py @@ -10,22 +10,22 @@ scheduler_brutal.HISTORY_DEPTH = 3 # Scheduling behaviour #-------------------------------------------------------------------------- def test_scheduling(): - cards = {id: Card("", "") for id in range(0, 10)} + cards = {str(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], + "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] + expected_priority = ["9", "6", "5", "7", "8", "4", "1", "3", "2", "0"] uut = UUT(cards, state) @@ -36,14 +36,14 @@ def test_scheduling(): # State update #-------------------------------------------------------------------------- def test_stateUpdate(): - cards = {0: Card("f", "b"), 1: Card("a", "b"), 2: Card("c", "d")} - state = {0: [1, 0, 1], 1: [1, 0, 0], 2: [0, 0, 1]} + cards = {"0": Card("f", "b"), "1": Card("a", "b"), "2": Card("c", "d")} + state = {"0": [1, 0, 1], "1": [1, 0, 0], "2": [0, 0, 1]} uut = UUT(cards, state) # Unknown IDs in the result are silently ignored - result = {0: 1, 1: 0, 3: 0} - expected_state = {0: [0, 1, 1], 1: [0, 0, 0], 2: [0, 1, None]} + result = {"0": 1, "1": 0, "3": 0} + expected_state = {"0": [0, 1, 1], "1": [0, 0, 0], "2": [0, 1, None]} uut.update(result) @@ -53,10 +53,10 @@ def test_stateUpdate(): # State corrections #-------------------------------------------------------------------------- def test_stateWhenCardsChanged(): - cards = {0: Card("f", "b"), 1: Card("a", "b")} + cards = {"0": Card("f", "b"), "1": Card("a", "b")} - initial_state = {0: [1, 0, 1], -1: [0, 0, 0]} - expected_state = {0: [1, 0, 1], 1: [None, None, None]} + initial_state = {"0": [1, 0, 1], "2": [0, 0, 0]} + expected_state = {"0": [1, 0, 1], "1": [None, None, None]} uut = UUT(cards, initial_state) @@ -65,10 +65,10 @@ def test_stateWhenCardsChanged(): def test_stateWhenHistoryDepthIncreased(): scheduler_brutal.HISTORY_DEPTH = 5 - cards = {0: Card("f", "b"), 1: Card("a", "b"), 2: Card("new", "new")} + cards = {"0": Card("f", "b"), "1": Card("a", "b"), "2": Card("new", "new")} - initial_state = {0: [1, 0, 1], 1: [0, 0, 0]} - expected_state = {0: [None, None, 1, 0, 1], 1: [None, None, 0, 0, 0], 2: [None] * 5} + initial_state = {"0": [1, 0, 1], "1": [0, 0, 0]} + expected_state = {"0": [None, None, 1, 0, 1], "1": [None, None, 0, 0, 0], "2": [None] * 5} uut = UUT(cards, initial_state) @@ -77,10 +77,10 @@ def test_stateWhenHistoryDepthIncreased(): def test_stateWhenHistoryDepthDecreased(): scheduler_brutal.HISTORY_DEPTH = 1 - cards = {0: Card("f", "b"), 1: Card("a", "b"), 2: Card("new", "new")} + cards = {"0": Card("f", "b"), "1": Card("a", "b"), "2": Card("new", "new")} - initial_state = {0: [1, 0, 0], 1: [0, 0, 1]} - expected_state = {0: [0], 1: [1], 2: [None]} + initial_state = {"0": [1, 0, 0], "1": [0, 0, 1]} + expected_state = {"0": [0], "1": [1], "2": [None]} uut = UUT(cards, initial_state) |