def test_can_update(self): game = GameOfLife(width=self.width, height=self.height, cell_size=1) game.grid = self.grid with open('homework03/steps.text') 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_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_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(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))