Added tests

1 files changed, 4 insertions, 0 deletions

diff --git a/day1/__init__.py b/day1/__init__.py
index bc44089..be17935 100644
--- a/day1/__init__.py
+++ b/day1/__init__.py
@@ -11,6 +11,8 @@ class Assignment(BaseAssignment):
         return list(super().read_input(example))
 
 class AssignmentOne(Assignment):
+    example_result = 7
+
     def run(self, input: List) -> int:
         result = 0
 
@@ -21,6 +23,8 @@ class AssignmentOne(Assignment):
 
 
 class AssignmentTwo(Assignment):
+    example_result = 5
+
     def run(self, input: List) -> int:
         new_input = [
             input[i - 2] + input[i - 1] + input[i]