class BasicTest(unittest.TestCase):

    def setUp(self):
        self.game = GameOfLife(10, 10)

    def test_set_alive(self):
        self.game.set_alive((0, 0))

        self.assertTrue(self.game.state[0][0])
        self.assertIn((0, 0), self.game.living_cells)

    def test_set_dead(self):
        self.game.set_alive((0, 0))
        self.assertTrue(self.game.state[0][0])
        self.assertIn((0, 0), self.game.living_cells)

        self.game.set_dead((0, 0))
        self.assertFalse(self.game.state[0][0])
        self.assertNotIn((0, 0), self.game.living_cells)

    def test_get_neighbours(self):
        neighbours = self.game.get_neighbours((0, 0))
        self.assertSetEqual(neighbours, set([(0, 1), (1, 0), (1, 1)]))

        neighbours = self.game.get_neighbours((0, 5))
        self.assertSetEqual(neighbours, set(
            [(0, 6), (0, 4), (1, 4), (1, 5), (1, 6)]))

        neighbours = self.game.get_neighbours((5, 0))
        self.assertSetEqual(neighbours, set(
            [(5, 1), (4, 0), (4, 1), (6, 0), (6, 1)]))

        neighbours = self.game.get_neighbours((5, 5))
        self.assertSetEqual(neighbours, set(
            [(5, 6), (5, 4), (4, 4), (4, 5), (4, 6), (6, 4), (6, 5), (6, 6)]))

    def test_life_over(self):
        self.game.set_alive((0, 1))
        self.game.set_alive((1, 0))
        self.game.set_alive((2, 0))

        self.game.update()

        self.assertEqual(len(self.game.living_cells), 2)
        self.assertIn((1, 0), self.game.living_cells)
        self.assertIn((1, 1), self.game.living_cells)

        self.game.update()

        self.assertTrue(self.game.life_over())

    def test_update(self):
        self.game.set_alive((0, 1))
        self.game.set_alive((1, 0))
        self.game.set_alive((2, 0))

        self.game.update()

        self.assertEqual(len(self.game.living_cells), 2)
        self.assertIn((1, 0), self.game.living_cells)
        self.assertIn((1, 1), self.game.living_cells)

    def test_reset(self):
        self.game.set_alive((0, 1))
        self.game.set_alive((1, 1))
        self.game.set_alive((2, 1))

        self.game.update()

        self.assertEqual(len(self.game.living_cells), 3)
        self.assertSetEqual(
            set([(1, 0), (1, 1), (1, 2)]), self.game.living_cells)

        self.game.reset()
        self.assertEqual(len(self.game.living_cells), 0)
        self.assertFalse(any(any(row) for row in self.game.state))
Esempio n. 2
0
class ShapeTest(unittest.TestCase):
    '''Test for various transitions
    Taken from http://pi.math.cornell.edu/~lipa/mec/lesson6.html
    '''

    def setUp(self):
        self.game = GameOfLife(10, 10)

    def test_triomino_1(self):
        '''Test for following transition
        □             □
        □ => □ □ □ => □ => so on
        □             □
        '''
        self.game.set_alive((0, 1))
        self.game.set_alive((1, 1))
        self.game.set_alive((2, 1))

        self.game.update()
        self.assertEqual(len(self.game.living_cells), 3)
        self.assertSetEqual(
            set([(1, 0), (1, 1), (1, 2)]), self.game.living_cells)

        self.game.update()
        self.assertEqual(len(self.game.living_cells), 3)
        self.assertSetEqual(
            set([(0, 1), (1, 1), (2, 1)]), self.game.living_cells)

        self.game.update()
        self.assertEqual(len(self.game.living_cells), 3)
        self.assertSetEqual(
            set([(1, 0), (1, 1), (1, 2)]), self.game.living_cells)

    def test_triomino_2(self):
        '''Test for following transitions
          □
         □   => □ => Nothing
        □
        '''
        self.game.set_alive((0, 2))
        self.game.set_alive((1, 1))
        self.game.set_alive((2, 0))

        self.game.update()
        self.assertEqual(len(self.game.living_cells), 1)
        self.assertSetEqual(set([(1, 1)]), self.game.living_cells)

        self.game.update()
        self.assertTrue(self.game.life_over())

    def test_tetromino_1(self):
        '''Test for following transitions
        □ □  => □ □ => Stable
        □ □     □ □
        '''
        self.game.set_alive((0, 0))
        self.game.set_alive((0, 1))
        self.game.set_alive((1, 0))
        self.game.set_alive((1, 1))

        self.game.update()
        self.assertEqual(len(self.game.living_cells), 4)
        self.assertSetEqual(
            set([(0, 0), (0, 1), (1, 0), (1, 1)]), self.game.living_cells)

        self.game.update()
        self.assertEqual(len(self.game.living_cells), 4)
        self.assertSetEqual(
            set([(0, 0), (0, 1), (1, 0), (1, 1)]), self.game.living_cells)

        self.game.update()
        self.assertEqual(len(self.game.living_cells), 4)
        self.assertSetEqual(
            set([(0, 0), (0, 1), (1, 0), (1, 1)]), self.game.living_cells)

    def test_tetromino_2(self):
        '''Test for following transitions
        □               □
        □ => □ □ □ => □   □ => stable
        □    □ □ □    □   □
        □               □
        '''
        self.game.set_alive((0, 1))
        self.game.set_alive((1, 1))
        self.game.set_alive((2, 1))
        self.game.set_alive((3, 1))

        self.game.update()
        self.assertEqual(len(self.game.living_cells), 6)
        self.assertSetEqual(
            set([(1, 0), (1, 1), (1, 2), (2, 0), (2, 1), (2, 2)]), self.game.living_cells)

        self.game.update()
        self.assertEqual(len(self.game.living_cells), 6)
        self.assertSetEqual(
            set([(1, 0), (2, 0), (0, 1), (3, 1), (1, 2), (2, 2)]), self.game.living_cells)

        self.game.update()
        self.assertEqual(len(self.game.living_cells), 6)
        self.assertSetEqual(
            set([(1, 0), (2, 0), (0, 1), (3, 1), (1, 2), (2, 2)]), self.game.living_cells)