formatting and cosmetic updates to 2020.

2020/days/06/test_day6.py

@@ -6,50 +6,33 @@ import day6 as aoc
 
 PUZZLE_DIR = pathlib.Path(__file__).parent
 
-#these test fixtures setup the test, mainly by reading the filename into a string in this simple case.
+
+# these test fixtures setup the test, mainly by reading the filename into a string in this simple case.
 @pytest.fixture
-def example1():
-    puzzle_input = (PUZZLE_DIR / "example1").read_text().strip()
+def example():
+    puzzle_input = (PUZZLE_DIR / "example.txt").read_text().strip()
     return aoc.parse(puzzle_input)
 
-@pytest.fixture
-def example2():
-    puzzle_input = (PUZZLE_DIR / "example2").read_text().strip()
-    return aoc.parse(puzzle_input)
 
 # @pytest.mark.skip(reason="Not implemented")
-def test_parse_example1(example1):
+def test_parse(example):
     """Test that input is parsed properly"""
-    assert example1 == [
-            [
-                ['a', 'b', 'c']
-                ],
-            [
-                ['a'],
-                ['b'],
-                ['c']
-                ],
-            [
-                ['a', 'b'],
-                ['a', 'c']
-            ],
-            [
-                ['a'],
-                ['a'],
-                ['a'],
-                ['a']
-            ],
-            [
-                ['b']
-            ]
-            ]
+    assert example == [
+        [["a", "b", "c"]],
+        [["a"], ["b"], ["c"]],
+        [["a", "b"], ["a", "c"]],
+        [["a"], ["a"], ["a"], ["a"]],
+        [["b"]],
+    ]
+
 
 # @pytest.mark.skip(reason="Not implemented")
-def test_part1_example1(example1):
+def test_part1(example):
     """Test part 1 on example input"""
-    assert aoc.part1(example1) == 11
+    assert aoc.part1(example) == 11
+
 
 # @pytest.mark.skip(reason="Not implemented")
-def test_part2_example2(example1):
+def test_part2(example):
     """Test part 2 on example input"""
-    assert aoc.part2(example1) == 6
+    assert aoc.part2(example) == 6