Day 5 part 2 solution.

day5/test.py

@@ -1,5 +1,5 @@
 import logging
-from solver import process_data
+from solver import sum_correct_middles, sum_incorrect_middles
 
 logger = logging.getLogger()
 logger.setLevel(logging.DEBUG)
@@ -37,10 +37,14 @@ test_data = r"""
 
 
 def test_main():
-    print("Beginning basic test")
-    result = process_data(iter(test_data.splitlines()))
+    print("Beginning correct middle sum test")
+    result = sum_correct_middles(iter(test_data.splitlines()))
     assert result == 143
-    print("Basic test passed")
+    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()