1 files changed, 8 insertions, 0 deletions

diff --git a/day20/test_init.py b/day20/test_init.py
index b3ccc44..24b94cb 100644
--- a/day20/test_init.py
+++ b/day20/test_init.py
@@ -25,3 +25,11 @@ def test_rotation():
 
     items = AssignmentOne.move(items, 4)
     assert items == [1, 2, -3, 4, 0, 3, -2]
+
+
+def test_get_nth_number_after_0():
+    items = [1, 2, -3, 4, 0, 3, -2]
+
+    assert AssignmentOne.get_nth_number_after_0(items, 1000) == 4
+    assert AssignmentOne.get_nth_number_after_0(items, 2000) == -3
+    assert AssignmentOne.get_nth_number_after_0(items, 3000) == 2