Added tests

1 files changed, 4 insertions, 0 deletions

diff --git a/day2/__init__.py b/day2/__init__.py
index 14229a9..d99a0be 100644
--- a/day2/__init__.py
+++ b/day2/__init__.py
@@ -17,6 +17,8 @@ class Assignment(BaseAssignment):
         return list(super().read_input(example))
 
 class AssignmentOne(Assignment):
+    example_result = 150
+
     depth = 0
     horizontal = 0
 
@@ -34,6 +36,8 @@ class AssignmentOne(Assignment):
 
 
 class AssignmentTwo(Assignment):
+    example_result = 900
+
     aim = 0
     depth = 0
     horizontal = 0