neuropercolation/test/test_factory.py

import sys
sys.path.append('../src')

from 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()