added more type hints

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

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)