def test_step_example1(self): grid = dec18.Grid(['.#.#.#', '...##.', '#....#', '..#...', '#.#..#', '####..']) expected = dec18.Grid(['..##..', '..##.#', '...##.', '......', '#.....', '#.##..']) self.assertEqual(str(expected), str(dec18.step(grid, 1)))
def test_broken_grid(self): grid = dec18.CornerLitGrid(['##.#.#', '...##.', '#....#', '..#...', '#.#..#', '####.#']) step = dec18.step(grid, 5) expected = dec18.CornerLitGrid(['##.###', '.##..#', '.##...', '.##...', '#.#...', '##...#']) self.assertEqual(str(expected), str(step)) self.assertEqual(17, step.count())