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()