31 lines
1 KiB
Python
31 lines
1 KiB
Python
|
import unittest
|
||
|
|
|
||
|
|
from solutions.day_01 import Solution
|
||
|
|
|
||
|
|
|
||
|
|
class Day01TestCase(unittest.TestCase):
|
||
|
|
def setUp(self):
|
||
|
|
self.solution = Solution()
|
||
|
|
|
||
|
|
def test_resolves_frequency(self):
|
||
|
|
freq_1 = '\n'.join(['+1', '+1', '+1'])
|
||
|
|
freq_2 = '\n'.join(['+1', '+1', '-2'])
|
||
|
|
freq_3 = '\n'.join(['-1', '-2', '-3'])
|
||
|
|
assert self.solution.solve(freq_1) == 3
|
||
|
|
assert self.solution.solve(freq_2) == 0
|
||
|
|
assert self.solution.solve(freq_3) == -6
|
||
|
|
|
||
|
|
def test_resolves_recurring_frequency(self):
|
||
|
|
freq_1 = '\n'.join(['+1', '-1'])
|
||
|
|
freq_2 = '\n'.join(['+3', '+3', '+4', '-2', '-4'])
|
||
|
|
freq_3 = '\n'.join(['-6', '+3', '+8', '+5', '-6'])
|
||
|
|
freq_4 = '\n'.join(['+7', '+7', '-2', '-7', '-4'])
|
||
|
|
assert self.solution.solve_again(freq_1) == 0
|
||
|
|
assert self.solution.solve_again(freq_2) == 10
|
||
|
|
assert self.solution.solve_again(freq_3) == 5
|
||
|
|
assert self.solution.solve_again(freq_4) == 14
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
unittest.main()
|