diff options
Diffstat (limited to 'src/scheduler_brutal_unittest.py')
| -rw-r--r-- | src/scheduler_brutal_unittest.py | 44 |
1 files changed, 18 insertions, 26 deletions
diff --git a/src/scheduler_brutal_unittest.py b/src/scheduler_brutal_unittest.py index 98de2f1..1ee6402 100644 --- a/src/scheduler_brutal_unittest.py +++ b/src/scheduler_brutal_unittest.py @@ -7,37 +7,29 @@ from card import Card scheduler_brutal.HISTORY_DEPTH = 3 #-------------------------------------------------------------------------- -# +# 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]} -#-------------------------------------------------------------------------- -# -#-------------------------------------------------------------------------- - -#-------------------------------------------------------------------------- -# -#-------------------------------------------------------------------------- + uut = UUT(cards, state) -#-------------------------------------------------------------------------- -# -#-------------------------------------------------------------------------- + result = {0: 1, 1: 0} + expected_state = {0: [0, 1, 1], 1: [0, 0, 0], 2: [0, 1, None]} -#-------------------------------------------------------------------------- -# -#-------------------------------------------------------------------------- + uut.update(result) -#-------------------------------------------------------------------------- -# -#-------------------------------------------------------------------------- + assert uut.getState() == expected_state #-------------------------------------------------------------------------- # 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], -1: [0, 0, 0]} + expected_state = {0: [1, 0, 1], 1: [None, None, None]} uut = UUT(cards, initial_state) @@ -46,10 +38,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) @@ -58,10 +50,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) |