import day22
import unittest


class TestDay22a(unittest.TestCase):

    def test_test_input(self):
        self.assertEqual(5587, day22.calc_a(test_input))

    def test_input(self):
        self.assertEqual(5266, day22.calc_a(input))


class TestDay22b(unittest.TestCase):

    def test_test_input(self):
       self.assertEqual(2511944, day22.calc_b(test_input))

    def test_input(self):
       self.assertEqual(2511895, day22.calc_b(input))

test_input = """..#
#..
..."""

input = """.########.....#...##.####
....#..#.#.##.###..#.##..
##.#.#..#.###.####.##.#..
####...#...####...#.##.##
..#...###.#####.....##.##
..#.##.######.#...###...#
.#....###..##....##...##.
##.##..####.#.######...##
#...#..##.....#..#...#..#
........#.##..###.#.....#
#.#..######.#.###..#...#.
.#.##.##..##.####.....##.
.....##..#....#####.#.#..
...#.#.#..####.#..###..#.
##.#..##..##....#####.#..
.#.#..##...#.#####....##.
.####.#.###.####...#####.
...#...######..#.##...#.#
#..######...#.####.#..#.#
...##..##.#.##.#.#.#....#
###..###.#..#.....#.##.##
..#....##...#..#..##..#..
.#.###.##.....#.###.#.###
####.##...#.#....#..##...
#.....#.#..#.##.#..###..#"""