advent-of-code/2017-python/tests/day_11_tests.py

22 lines
610 B
Python

import unittest
from solutions.day_11 import Solution
class Day11TestCase(unittest.TestCase):
def setUp(self):
self.solution = Solution()
def test_distance(self):
assert self.solution.solve("ne,ne,ne") == 3
assert self.solution.solve("ne,ne,sw,sw") == 0
assert self.solution.solve("ne,ne,s,s") == 2
assert self.solution.solve("se,sw,se,sw,sw") == 3
def test_furthest_away(self):
assert self.solution.solve_again("ne,ne,sw,sw") == 2
assert self.solution.solve_again("se,sw,se,sw,sw") == 3
if __name__ == "__main__":
unittest.main()