advent-of-code/2018-python/tests/day_11_tests.py
2021-11-01 16:45:03 +01:00

25 lines
776 B
Python

import unittest
from solutions.day_11 import Solution
class Day11TestCase(unittest.TestCase):
def setUp(self):
self.solution = Solution()
def test_power_level(self):
assert self.solution.get_power_level(122, 79, 57) == -5
assert self.solution.get_power_level(217, 196, 39) == 0
assert self.solution.get_power_level(101, 153, 71) == 4
def test_largest_square_size3x3(self):
assert self.solution.solve('18') == (33, 45)
assert self.solution.solve('42') == (21, 61)
def test_largest_square_any_size(self):
assert self.solution.solve_again('18') == (90, 269, 16)
assert self.solution.solve_again('42') == (232, 251, 12)
if __name__ == '__main__':
unittest.main()