neuropercolation/test/test_state.py
2019-02-17 08:19:04 +01:00

21 lines
482 B
Python

import sys
sys.path.append('../src')
import cellular_automaton.ca_cell_state as cs
import unittest
class TestState(unittest.TestCase):
def test_state_slots(self):
cell_state = cs.CellState([0])
cell_state.set_status_of_iteration([1], 0)
cell_state.set_status_of_iteration([2], 1)
cell_state.set_status_of_iteration([3], 2)
self.assertEqual(cell_state.get_status_of_iteration(0), [3])
if __name__ == '__main__':
unittest.main()