def test_next_state(self): self.assertEqual(get_state(next_state(build_map(self.d))), self.results[0]) self.assertEqual(get_state(next_state(build_map(self.results[0]))), self.results[1]) self.assertEqual(get_state(next_state(build_map(self.results[1]))), self.results[2]) self.assertEqual( count_lights_on(next_state(build_map(self.results[1]))), 4)
def test_next_state(self): grid = [ [False, True, False, True, False, True], [False, False, False, True, True, False], [True, False, False, False, False, True], [False, False, True, False, False, False], [True, False, True, False, False, True], [True, True, True, True, False, False] ] # Test after 1 step expected_grid = [ [False, False, True, True, False, False], [False, False, True, True, False, True], [False, False, False, True, True, False], [False, False, False, False, False, False], [True, False, False, False, False, False], [True, False, True, True, False, False] ] r = day18.next_state(grid) self.assertEqual(r, expected_grid) # Test after 4 steps expected_grid = [ [False, False, False, False, False, False], [False, False, False, False, False, False], [False, False, True, True, False, False], [False, False, True, True, False, False], [False, False, False, False, False, False], [False, False, False, False, False, False] ] r = day18.next_state(r) r = day18.next_state(r) r = day18.next_state(r) self.assertEqual(r, expected_grid)
def test_next_state_part2(self): grid = [ [True, True, False, True, False, True], [False, False, False, True, True, False], [True, False, False, False, False, True], [False, False, True, False, False, False], [True, False, True, False, False, True], [True, True, True, True, False, True] ] # Test after 1 step expected_grid = [ [True, False, True, True, False, True], [True, True, True, True, False, True], [False, False, False, True, True, False], [False, False, False, False, False, False], [True, False, False, False, True, False], [True, False, True, True, True, True] ] r = day18.next_state(grid, True) self.assertEqual(r, expected_grid) # Test after 5 steps expected_grid = [ [True, True, False, True, True, True], [False, True, True, False, False, True], [False, True, True, False, False, False], [False, True, True, False, False, False], [True, False, True, False, False, False], [True, True, False, False, False, True] ] for i in range(4): r = day18.next_state(r, True) self.assertEqual(r, expected_grid)
def test_next_state(self): self.assertEqual(get_state(next_state(build_map(self.d))), self.results[0]) self.assertEqual(get_state(next_state(build_map(self.results[0]))), self.results[1]) self.assertEqual(get_state(next_state(build_map(self.results[1]))), self.results[2]) self.assertEqual(count_lights_on(next_state(build_map(self.results[1]))), 4)