Coverage for shared/test_state.py: 100%

30 statements  

« prev     ^ index     » next       coverage.py v7.8.0, created at 2025-05-12 11:24 -0400

1from birdbrain_state import BirdbrainState 

2 

3def test_state(): 

4 state = BirdbrainState() 

5 

6 for pixel in state.display_map: 

7 assert pixel == 0 

8 

9 assert state.display_map[0] == 0 

10 assert state.display_map[18] == 0 

11 

12 state.set_pixel(1, 1, 1) 

13 state.set_pixel(4, 4, 1) 

14 

15 assert state.display_map[0] == 1 

16 assert state.display_map[18] == 1 

17 assert state.display_map[1] == 0 

18 assert state.display_map[19] == 0 

19 

20 s = state.display_map_normalize() 

21 

22 assert s[0] == "true" 

23 assert s[18] == "true" 

24 assert s[1] == "false" 

25 assert s[19] == "false" 

26 

27 assert state.display_map_as_string() == "true/false/false/false/false/false/false/false/false/false/false/false/false/false/false/false/false/false/true/false/false/false/false/false/false" 

28 

29def test_display_map_as_string_with_list(): 

30 state = BirdbrainState() 

31 

32 list = [ 0,1,0,1,0,1,0,1,0,1,0,1,0,1,0,1,0,1,0,1,0,1,0,1,0 ] 

33 

34 assert state.display_map_as_string(list)[0:11] == "false/true/" 

35 

36def test_state_using_true_and_false(): 

37 state = BirdbrainState() 

38 

39 state.set_pixel(1, 1, False) 

40 state.set_pixel(4, 4, True) 

41 

42 s = state.display_map_normalize() 

43 

44 assert s[0] == "false" 

45 assert s[18] == "true"