51 lines
792 B
Python
51 lines
792 B
Python
import logging
|
|
from solver import sum_correct_middles, sum_incorrect_middles
|
|
|
|
logger = logging.getLogger()
|
|
logger.setLevel(logging.DEBUG)
|
|
|
|
test_data = r"""
|
|
47|53
|
|
97|13
|
|
97|61
|
|
97|47
|
|
75|29
|
|
61|13
|
|
75|53
|
|
29|13
|
|
97|29
|
|
53|29
|
|
61|53
|
|
97|53
|
|
61|29
|
|
47|13
|
|
75|47
|
|
97|75
|
|
47|61
|
|
75|61
|
|
47|29
|
|
75|13
|
|
53|13
|
|
|
|
75,47,61,53,29
|
|
97,61,53,29,13
|
|
75,29,13
|
|
75,97,47,61,53
|
|
61,13,29
|
|
97,13,75,29,47
|
|
"""
|
|
|
|
|
|
def test_main():
|
|
print("Beginning correct middle sum test")
|
|
result = sum_correct_middles(iter(test_data.splitlines()))
|
|
assert result == 143
|
|
print("Correct middle sum test passed")
|
|
print("Beginning incorrect middle sum test")
|
|
result = sum_incorrect_middles(iter(test_data.splitlines()))
|
|
assert result == 123
|
|
print("Incorrect middle sum test passed")
|
|
|
|
if __name__ == '__main__':
|
|
test_main()
|