tests.py

import unittest

from gameoflife import life_check, run_mutations


class GameofLifeTest(unittest.TestCase):
    def setUp(self):
        # Set up a grid with a blinker
        grid = [[0]*8 for x in range(8)]
        grid[4][4] = 1
        grid[4][5] = 1
        grid[4][6] = 1

    def test_life_check(self):

        # Set up a grid with a blinker
        grid = [[0]*8 for x in range(8)]
        grid[4][4] = 1
        grid[4][5] = 1
        grid[4][6] = 1

        self.assertTrue(life_check(grid, 4, 5))
        self.assertTrue(life_check(grid, 3, 5))
        self.assertFalse(life_check(grid, 4, 4))
        self.assertFalse(life_check(grid, 4, 6))

    def test_mutate(self):
        # Set up a grid with a blinker
        grid = [[0]*8 for x in range(8)]
        grid[4][4] = 1
        grid[4][5] = 1
        grid[4][6] = 1

        run_mutations(1, grid)

        self.assertTrue(life_check(grid, 4, 4))
        self.assertTrue(life_check(grid, 4, 6))
        self.assertFalse(life_check(grid, 4, 5))
        self.assertFalse(life_check(grid, 3, 5))

if __name__ == '__main__':
    unittest.main()