neuropercolation/test/test_cell_state.py
2019-02-17 08:19:05 +01:00

43 lines
1.6 KiB
Python

import sys
sys.path.append('../src')
import cellular_automaton.ca_cell_state as cs
import unittest
class TestCellState(unittest.TestCase):
def test_get_state_with_overflow(self):
cell_state = cs.CellState(initial_state=(0,))
cell_state.set_state_of_iteration(new_state=(1,), iteration=0)
self.assertEqual(tuple(cell_state.get_state_of_iteration(2)), (1,))
def test_set_state_with_overflow(self):
cell_state = cs.CellState(initial_state=(0,))
cell_state.set_state_of_iteration(new_state=(1,), iteration=2)
self.assertEqual(tuple(cell_state.get_state_of_iteration(0)), (1,))
def test_set_state_does_not_effect_all_slots(self):
cell_state = cs.CellState(initial_state=(0,))
cell_state.set_state_of_iteration(new_state=(1,), iteration=0)
self.assertEqual(tuple(cell_state.get_state_of_iteration(1)), (0,))
def test_redraw_state_on_change(self):
cell_state = cs.CellState(initial_state=(0,), draw_first_state=False)
cell_state.set_state_of_iteration(new_state=(1,), iteration=0)
self.assertTrue(cell_state.is_set_for_redraw())
def test_redraw_state_on_nochange(self):
cell_state = cs.CellState(initial_state=(0,), draw_first_state=False)
cell_state.set_state_of_iteration(new_state=(0,), iteration=0)
self.assertFalse(cell_state.is_set_for_redraw())
def test_active_state_after_set(self):
cell_state = cs.CellState(initial_state=(0,), draw_first_state=False)
cell_state.set_state_of_iteration(new_state=(1,), iteration=0)
self.assertFalse(cell_state.is_active(1))
if __name__ == '__main__':
unittest.main()