summaryrefslogtreecommitdiffstats
path: root/src/scheduler_brutal_unittest.py
blob: 98de2f1c8bdf7fcb8f253172151e9c30b6c6859a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
import pytest
import scheduler_brutal
from scheduler_brutal import SchedulerBrutal as UUT
from card import Card

# Force HISTORY_DEPTH to simplify testing
scheduler_brutal.HISTORY_DEPTH = 3

#--------------------------------------------------------------------------
# 
#--------------------------------------------------------------------------

#--------------------------------------------------------------------------
# 
#--------------------------------------------------------------------------

#--------------------------------------------------------------------------
# 
#--------------------------------------------------------------------------

#--------------------------------------------------------------------------
# 
#--------------------------------------------------------------------------

#--------------------------------------------------------------------------
# 
#--------------------------------------------------------------------------

#--------------------------------------------------------------------------
# 
#--------------------------------------------------------------------------

#--------------------------------------------------------------------------
# State corrections
#--------------------------------------------------------------------------
def test_stateWhenCardsChanged():
    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]}

    uut = UUT(cards, initial_state)

    assert uut.getState() == expected_state

def test_stateWhenHistoryDepthIncreased():
    scheduler_brutal.HISTORY_DEPTH = 5

    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}

    uut = UUT(cards, initial_state)

    assert uut.getState() == expected_state

def test_stateWhenHistoryDepthDecreased():
    scheduler_brutal.HISTORY_DEPTH = 1

    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]}

    uut = UUT(cards, initial_state)

    assert uut.getState() == expected_state