32 lines
1 KiB
Python
32 lines
1 KiB
Python
|
import unittest
|
||
|
|
|
||
|
|
from solutions.day_02 import Solution
|
||
|
|
|
||
|
|
|
||
|
|
class Day2TestCase(unittest.TestCase):
|
||
|
|
def setUp(self):
|
||
|
|
self.solution = Solution()
|
||
|
|
|
||
|
|
def test_calculates_row_difference(self):
|
||
|
|
assert self.solution.get_diff([5, 1, 9, 5]) == 8
|
||
|
|
assert self.solution.get_diff([7, 5, 3]) == 4
|
||
|
|
assert self.solution.get_diff([2, 4, 6, 8]) == 6
|
||
|
|
|
||
|
|
def test_calculates_checksum(self):
|
||
|
|
puzzle_input = '\n'.join(['5 1 9 5', '7 5 3', '2 4 6 8'])
|
||
|
|
assert self.solution.solve(puzzle_input) == 18
|
||
|
|
|
||
|
|
def test_calculates_row_even_divisible(self):
|
||
|
|
puzzle_input = '\n'.join(['5 9 2 8', '9 4 7 3', '3 8 6 5'])
|
||
|
|
assert self.solution.get_even_divisible([5, 9, 2, 8]) == 4
|
||
|
|
assert self.solution.get_even_divisible([9, 4, 7, 3]) == 3
|
||
|
|
assert self.solution.get_even_divisible([3, 8, 6, 5]) == 2
|
||
|
|
|
||
|
|
def test_calculates_row_result_sum(self):
|
||
|
|
puzzle_input = '\n'.join(['5 9 2 8', '9 4 7 3', '3 8 6 5'])
|
||
|
|
assert self.solution.solve_again(puzzle_input) == 9
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
unittest.main()
|