def test_get_neighbours_for_left_side(self):
game = life_proto.GameOfLife(width=self.width,
height=self.height,
cell_size=1)
game.grid = self.grid
neighbours = game.get_neighbours((2, 0))
self.assertEqual(5, len(neighbours))
self.assertEqual(2, sum(neighbours))
def test_get_neighbours_for_lower_right_corner(self):
game = life_proto.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_can_update(self):
game = life_proto.GameOfLife(width=self.width,
height=self.height,
cell_size=1)
game.grid = self.grid
tests_dir = os.path.dirname(__file__)
steps_path = os.path.join(tests_dir, "steps.txt")
with open(steps_path) 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_can_create_a_random_grid(self):
game = life_proto.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 = life_proto.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)