class LatticeLivenessTests(unittest.TestCase):
def setUp(self):
self.lattice = Lattice(4)
def test_cells_are_dead_by_default(self):
for x in range(self.lattice.size):
for y in range(self.lattice.size):
self.assertTrue(self.lattice.is_dead(x, y), "x = {}, y = {}".format(x, y))
def test_cell_is_live_after_make_live(self):
self.lattice.make_live(2, 3)
self.assertTrue(self.lattice.is_live(2, 3))
def test_cell_is_dead_after_making_live_cell_dead(self):
self.lattice.make_live(2, 3)
self.lattice.make_dead(2, 3)
self.assertTrue(self.lattice.is_dead(2, 3))
def test_dead_cell_becomes_live_after_toggle(self):
self.lattice.make_dead(1, 2)
self.lattice.toggle_liveness(1, 2)
self.assertTrue(self.lattice.is_live(1, 2))
def test_live_cell_becomes_dead_after_toggle(self):
self.lattice.make_live(1, 2)
self.lattice.toggle_liveness(1, 2)
self.assertTrue(self.lattice.is_dead(1, 2))
class Game:
def __init__(self, size):
self._lattice = Lattice(size)
@staticmethod
def from_string(str, dead_symbol=' ', live_symbol='x'):
lattice = Lattice.from_string(str, dead_symbol, live_symbol)
game = Game(lattice.size)
game._lattice = lattice
return game
@property
def size(self):
return self._lattice.size
def make_step(self):
new_lattice = Lattice(self.size)
for x in range(self.size):
for y in range(self.size):
if self._should_become_live(x, y):
new_lattice.make_live(x, y)
self._lattice = new_lattice
def _should_become_live(self, x, y):
num_of_live_neighbours = self._lattice.get_num_of_live_neighbours(x, y)
if self._lattice.is_live(x, y) and num_of_live_neighbours < 2:
return False
elif self._lattice.is_live(x, y) and 2 <= num_of_live_neighbours <= 3:
return True
elif self._lattice.is_live(x, y) and num_of_live_neighbours > 3:
return False
elif self._lattice.is_dead(x, y) and num_of_live_neighbours == 3:
return True
return False
def __eq__(self, other):
return self._lattice == other._lattice
def __ne__(self, other):
return self._lattice != other._lattice
def __repr__(self):
return repr(self._lattice)
def __getattr__(self, name):
try:
return getattr(self._lattice, name)
except AttributeError as e:
raise AttributeError(str(e).replace('Lattice', 'Game', 1)) from e