69 lines
2.6 KiB
Python
69 lines
2.6 KiB
Python
import sys
|
|
sys.path.append('../src')
|
|
|
|
from cellular_automaton.cellular_automaton import *
|
|
import unittest
|
|
import mock
|
|
|
|
|
|
class TestFac(CAFactory):
|
|
@staticmethod
|
|
def make_cells(dimension, state_class):
|
|
return CAFactory._make_cells(dimension, state_class)
|
|
|
|
@staticmethod
|
|
def apply_neighborhood(cells, neighborhood, dimension):
|
|
return CAFactory._apply_neighborhood_to_cells(cells, neighborhood, dimension)
|
|
|
|
|
|
class TestCAFactory(unittest.TestCase):
|
|
def setUp(self):
|
|
self._neighborhood = MooreNeighborhood(EdgeRule.IGNORE_EDGE_CELLS)
|
|
|
|
def test_make_ca_calls_correct_methods(self):
|
|
with mock.patch.object(CAFactory, '_make_cells', return_value={1: True}) as m1:
|
|
with mock.patch.object(CAFactory, '_apply_neighborhood_to_cells') as m2:
|
|
CAFactory.make_cellular_automaton([10], self._neighborhood, CellState, Rule())
|
|
m1.assert_called_once_with([10], CellState)
|
|
m2.assert_called_once_with({1: True}, self._neighborhood, [10])
|
|
|
|
def test_make_ca_returns_correct_values(self):
|
|
with mock.patch.object(CAFactory, '_make_cells', return_value={1: True}):
|
|
with mock.patch.object(CAFactory, '_apply_neighborhood_to_cells'):
|
|
ca = CAFactory.make_cellular_automaton([10], self._neighborhood, CellState, Rule())
|
|
self.assertIsInstance(ca, CellularAutomatonState)
|
|
self.assertEqual(tuple(ca.cells.values()), (True, ))
|
|
|
|
def test_1dimension_coordinates(self):
|
|
c = TestFac.make_cells([3], CellState)
|
|
self.assertEqual(list(c.keys()), [(0,), (1,), (2,)])
|
|
|
|
def test_2dimension_coordinates(self):
|
|
c = TestFac.make_cells([2, 2], CellState)
|
|
self.assertEqual(list(c.keys()), [(0, 0), (0, 1), (1, 0), (1, 1)])
|
|
|
|
def test_3dimension_coordinates(self):
|
|
c = TestFac.make_cells([2, 2, 2], CellState)
|
|
self.assertEqual(list(c.keys()), [(0, 0, 0), (0, 0, 1), (0, 1, 0), (0, 1, 1),
|
|
(1, 0, 0), (1, 0, 1), (1, 1, 0), (1, 1, 1)])
|
|
|
|
def test_apply_neighborhood(self):
|
|
cells = TestFac.make_cells([3, 3], CellState)
|
|
TestFac.apply_neighborhood(cells, self._neighborhood, [3, 3])
|
|
|
|
neighbors = self.__create_neighbor_list_of_cell((1, 1), cells)
|
|
|
|
self.assertEqual(set(neighbors), set(cells[(1, 1)].neighbor_states))
|
|
|
|
@staticmethod
|
|
def __create_neighbor_list_of_cell(cell_id, cells):
|
|
neighbors = []
|
|
for c in cells.values():
|
|
if c != cells[cell_id]:
|
|
neighbors.append(c.state)
|
|
return neighbors
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|