59 lines
2.9 KiB
Python
59 lines
2.9 KiB
Python
import sys
|
|
sys.path.append('../src')
|
|
|
|
import cellular_automaton.ca_neighborhood as csn
|
|
import unittest
|
|
|
|
|
|
class TestNeighborhood(unittest.TestCase):
|
|
@staticmethod
|
|
def check_neighbors(neighborhood, neighborhood_sets, dimension=(3, 3)):
|
|
for neighborhood_set in neighborhood_sets:
|
|
neighbors = neighborhood.calculate_cell_neighbor_coordinates(neighborhood_set[0], dimension)
|
|
if neighborhood_set[1] != neighbors:
|
|
print("\nrel_n:", neighborhood._rel_neighbors)
|
|
print("\nWrong neighbors (expected, real): ", (neighborhood_set[1]), neighbors)
|
|
return False
|
|
return True
|
|
|
|
def test_ignore_missing_neighbors(self):
|
|
neighborhood = csn.MooreNeighborhood(csn.EdgeRule.IGNORE_MISSING_NEIGHBORS_OF_EDGE_CELLS)
|
|
n00 = [[0, 0], [[1, 0], [0, 1], [1, 1]]]
|
|
n11 = [[1, 1], [[0, 0], [1, 0], [2, 0], [0, 1], [2, 1], [0, 2], [1, 2], [2, 2]]]
|
|
n22 = [[2, 2], [[1, 1], [2, 1], [1, 2]]]
|
|
self.assertTrue(self.check_neighbors(neighborhood, [n00, n11, n22]))
|
|
|
|
def test_ignore_edge_cells(self):
|
|
neighborhood = csn.MooreNeighborhood(csn.EdgeRule.IGNORE_EDGE_CELLS)
|
|
n00 = [[0, 0], []]
|
|
n11 = [[1, 1], [[0, 0], [1, 0], [2, 0], [0, 1], [2, 1], [0, 2], [1, 2], [2, 2]]]
|
|
n22 = [[2, 2], []]
|
|
self.assertTrue(self.check_neighbors(neighborhood, [n00, n11, n22]))
|
|
|
|
def test_cyclic_dimensions(self):
|
|
neighborhood = csn.MooreNeighborhood(csn.EdgeRule.FIRST_AND_LAST_CELL_OF_DIMENSION_ARE_NEIGHBORS)
|
|
n00 = [[0, 0], [[2, 2], [0, 2], [1, 2], [2, 0], [1, 0], [2, 1], [0, 1], [1, 1]]]
|
|
n11 = [[1, 1], [[0, 0], [1, 0], [2, 0], [0, 1], [2, 1], [0, 2], [1, 2], [2, 2]]]
|
|
n22 = [[2, 2], [[1, 1], [2, 1], [0, 1], [1, 2], [0, 2], [1, 0], [2, 0], [0, 0]]]
|
|
self.assertTrue(self.check_neighbors(neighborhood, [n00, n11, n22]))
|
|
|
|
def test_von_neumann_r1(self):
|
|
neighborhood = csn.VonNeumannNeighborhood(csn.EdgeRule.FIRST_AND_LAST_CELL_OF_DIMENSION_ARE_NEIGHBORS)
|
|
n1 = [[1, 1], [[1, 0], [0, 1], [2, 1], [1, 2]]]
|
|
self.assertTrue(self.check_neighbors(neighborhood, [n1]))
|
|
|
|
def test_von_neumann_r2(self):
|
|
neighborhood = csn.VonNeumannNeighborhood(csn.EdgeRule.FIRST_AND_LAST_CELL_OF_DIMENSION_ARE_NEIGHBORS, range_=2)
|
|
n1 = [[2, 2], [[2, 0], [1, 1], [2, 1], [3, 1], [0, 2], [1, 2], [3, 2], [4, 2], [1, 3], [2, 3], [3, 3], [2, 4]]]
|
|
self.assertTrue(self.check_neighbors(neighborhood, [n1], dimension=[5, 5]))
|
|
|
|
def test_von_neumann_d3(self):
|
|
neighborhood = csn.VonNeumannNeighborhood(csn.EdgeRule.FIRST_AND_LAST_CELL_OF_DIMENSION_ARE_NEIGHBORS,
|
|
dimension=3)
|
|
n1 = [[1, 1, 1], [[1, 1, 0], [1, 0, 1], [0, 1, 1], [2, 1, 1], [1, 2, 1], [1, 1, 2]]]
|
|
self.assertTrue(self.check_neighbors(neighborhood, [n1], dimension=[3, 3, 3]))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|