-
Notifications
You must be signed in to change notification settings - Fork 0
/
test_stop_and_go_FSM.py
44 lines (35 loc) · 1.59 KB
/
test_stop_and_go_FSM.py
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
import unittest
from stop_and_go_FSM import doing_pollution
class TestStopAndGoFSM(unittest.TestCase):
def test_keep_stop_without_trucks_waiting(self):
state, pollution = doing_pollution("stop", False, 0)
self.assertFalse(pollution)
self.assertEqual(state, "stop")
def test_keep_going_without_trucks_waiting(self):
state, pollution = doing_pollution("go", False, 0)
self.assertFalse(pollution)
self.assertEqual(state, "go")
def test_keep_stop_with_trucks_waiting(self):
state, pollution = doing_pollution("stop", True, 0)
self.assertTrue(pollution)
self.assertEqual(state, "stop")
def test_keep_going_with_trucks_waiting(self):
state, pollution = doing_pollution("go", True, 0)
self.assertFalse(pollution)
self.assertEqual(state, "go")
def test_stop_turn_to_go_without_trucks_waiting(self):
state, pollution = doing_pollution("stop", False, 1)
self.assertFalse(pollution)
self.assertEqual(state, "go")
def test_go_turn_to_stop_without_trucks_waiting(self):
state, pollution = doing_pollution("go", False, 1)
self.assertFalse(pollution)
self.assertEqual(state, "stop")
def test_stop_turn_to_go_with_trucks_waiting(self):
state, pollution = doing_pollution("stop", True, 1)
self.assertFalse(pollution)
self.assertEqual(state, "go")
def test_go_turn_to_stop_with_trucks_waiting(self):
state, pollution = doing_pollution("go", True, 1)
self.assertFalse(pollution)
self.assertEqual(state, "stop")