Implement 2019 day 16 part 2

2019/tests/day16.py

@@ -1,15 +0,0 @@
-import pytest
-
-from aoc2019.day16 import simulate
-
-
-@pytest.mark.parametrize('data,correct', [
-    ('80871224585914546619083218645595', '24176176'),
-    ('19617804207202209144916044189917', '73745418'),
-    ('69317163492948606335995924319873', '52432133'),
-])
-def test_sample_part1(data: str, correct: str):
-    numbers = [int(c) for c in data]
-
-    assert simulate(numbers) == correct
-    pass

2019/tests/test_day16.py

@@ -0,0 +1,25 @@
+import pytest
+
+from aoc2019.day16 import simulate, simulate2
+
+
+@pytest.mark.parametrize('data,correct', [
+    ('80871224585914546619083218645595', '24176176'),
+    ('19617804207202209144916044189917', '73745418'),
+    ('69317163492948606335995924319873', '52432133'),
+])
+def test_sample_part1(data: str, correct: str):
+    numbers = [int(c) for c in data]
+
+    assert simulate(numbers) == correct
+
+
+@pytest.mark.parametrize('data,correct', [
+    ('03036732577212944063491565474664', '84462026'),
+    ('02935109699940807407585447034323', '78725270'),
+    ('03081770884921959731165446850517', '53553731'),
+])
+def test_sample_part2(data: str, correct: str):
+    numbers = [int(c) for c in data]
+
+    assert simulate2(numbers) == correct