From 22987bcef81fa1d016ff444f0434679ead1297fe Mon Sep 17 00:00:00 2001
From: Arne Wischer <arne.wischer@dataport.de>
Date: Fri, 6 Dec 2019 15:46:51 +0100
Subject: [PATCH] added more type hints

---
 days/day2.py | 7 +++++--
 days/day5.py | 2 ++
 2 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/days/day2.py b/days/day2.py
index 18c218f..8e08ab9 100644
--- a/days/day2.py
+++ b/days/day2.py
@@ -1,5 +1,6 @@
 import unittest
 import os
+from typing import *
 from .comp import OpcodeComputer
 
 
@@ -8,14 +9,16 @@ class Day2(unittest.TestCase):
 
     def test_day2a(self):
         with open(self.inputfile) as fp:
-            code = [int(k) for k in fp.readline().split(',')]
+            code: List[int] = [int(k) for k in fp.readline().split(',')]
         code[1] = 12
         code[2] = 2
         self.assertEqual(OpcodeComputer().process_op(code), 5098658)
 
     def test_day2b(self):
         with open(self.inputfile) as fp:
-            file_code = [int(k) for k in fp.readline().split(',')]
+            file_code: List[int] = [int(k) for k in fp.readline().split(',')]
+        a: int = 0
+        b: int = 0
         for a, b in [(x, y) for x in range(100) for y in range(100)]:
             code = list(file_code)
             code[1] = a
diff --git a/days/day5.py b/days/day5.py
index 6ebb709..abb3272 100644
--- a/days/day5.py
+++ b/days/day5.py
@@ -13,6 +13,7 @@ class Day5(unittest.TestCase):
         output_val = []
 
         sut = comp.OpcodeComputer()
+        # replace input() and print() to simulate user behavior
         comp.input = lambda: 1
         comp.print = lambda s: output_val.append(s)
 
@@ -23,6 +24,7 @@ class Day5(unittest.TestCase):
         output_val = []
 
         sut = comp.OpcodeComputer()
+        # replace input() and print() to simulate user behavior
         comp.input = lambda: 5
         comp.print = lambda s: output_val.append(s)