class GameOfLifeTest(unittest.TestCase):
def setUp(self):
self.game = GameOfLife({Vec2D(0, 0): Cell(True), Vec2D(0, 1): Cell(True),
Vec2D(1, 0): Cell(True), Vec2D(3, 3): Cell(True)})
def test_alive_cell_when_game_is_set_up(self):
self.assertEqual(self.game.is_alive(Vec2D(1, 0)), True)
self.assertEqual(self.game.is_alive(Vec2D(1, 1)), False)
def test_number_of_alive_neighbours(self):
self.assertEqual(self.game.number_of_alive_neighbours(Vec2D(0, 0)), 2)
self.assertEqual(self.game.number_of_alive_neighbours(Vec2D(1, 1)), 3)
self.assertEqual(self.game.number_of_alive_neighbours(Vec2D(4, 0)), 0)
def test_neighbours_of(self):
self.assertEqual(sorted(self.game.neighbours_of(Vec2D(0, 0))), sorted([Vec2D(-1, -1), Vec2D(-1, 0), Vec2D(-1, 1),
Vec2D(0, -1), Vec2D(0, 1),
Vec2D(1, -1), Vec2D(1, 0), Vec2D(1, 1)]))
def test_will_be_born(self):
self.assertEqual(self.game.will_be_born(Vec2D(1, 1)), True)
self.assertEqual(self.game.will_be_born(Vec2D(0, 1)), False)
def test_will_servive(self):
self.assertEqual(self.game.will_servive(Vec2D(0, 1)), True)
self.assertEqual(self.game.will_servive(Vec2D(3, 3)), False)
def test_stable_cells(self):
stable_cells = self.game.stable_cells()
self.assertEqual(sorted(stable_cells.keys()), sorted([Vec2D(0, 0), Vec2D(0, 1), Vec2D(1, 0)]))
self.assertEqual(stable_cells[Vec2D(0, 0)].new_born, False)
def test_new_born_cells(self):
new_born_cells = self.game.new_born()
self.assertEqual(sorted(new_born_cells.keys()), [Vec2D(1, 1)])
self.assertEqual(new_born_cells[Vec2D(1, 1)].new_born, True)
def test_next_generation(self):
next_generation = self.game.next_generation()
self.assertEqual(sorted(next_generation.keys()), sorted([Vec2D(0, 0), Vec2D(0, 1), Vec2D(1, 0), Vec2D(1, 1)]))
