def test_should_return_correct_set_of_alive_neighbours_of_cell(self): # GIVEN seed = set([(-1, -1), (-1, 0)]) game_of_life = GameOfLife(seed) cell = (0, 0) expected_neighbours = set([(-1, -1), (-1, 0)]) # WHEN alive_neighbours = game_of_life.get_alive_neighbours(cell) # THEN self.assertEqual(expected_neighbours, alive_neighbours)