def test_can_update(self):
game = GameOfLife(width=self.width, height=self.height, cell_size=1)
game.grid = self.grid
with open('steps.txt') as f:
steps = json.load(f)
num_updates = 0
for step in sorted(steps.keys(), key=int):
with self.subTest(step=step):
for _ in range(int(step) - num_updates):
game.grid = game.get_next_generation()
num_updates += 1
self.assertEqual(steps[step], game.grid)
def test_get_neighbours_for_right_side(self):
game = GameOfLife(width=self.width, height=self.height, cell_size=1)
game.grid = self.grid
neighbours = game.get_neighbours((2, 7))
self.assertEqual(5, len(neighbours))
self.assertEqual(2, sum(neighbours))
def test_get_neighbours_for_lower_right_corner(self):
game = GameOfLife(width=self.width, height=self.height, cell_size=1)
game.grid = self.grid
neighbours = game.get_neighbours((5, 7))
self.assertEqual(3, len(neighbours))
self.assertEqual(1, sum(neighbours))
def test_get_neighbours(self):
game = GameOfLife(width=self.width, height=self.height, cell_size=1)
game.grid = self.grid
neighbours = game.get_neighbours((2, 3))
self.assertEqual(8, len(neighbours))
self.assertEqual(4, sum(neighbours))
def test_can_create_a_random_grid(self):
game = GameOfLife(width=3, height=3, cell_size=1)
random.seed(12345)
grid = game.create_grid(randomize=True)
self.assertEqual([[1, 0, 1], [1, 0, 1], [1, 0, 1]], grid)
def test_can_create_an_empty_grid(self):
game = GameOfLife(width=3, height=3, cell_size=1)
grid = game.create_grid(randomize=False)
self.assertEqual([[0, 0, 0], [0, 0, 0], [0, 0, 0]], grid)
