2015: Add day 1 part 2

2015/pyproject.toml

@@ -12,4 +12,5 @@ dependencies = [
 disable = [
     "missing-module-docstring",
     "missing-function-docstring",
+    "multiple-statements",
 ]

2015/test_day01.py

@@ -2,10 +2,23 @@
 
 
 def test_part01():
-    input_data = read_input("day01-input.txt")
+    directions = read_input("day01-input.txt")
 
-    up = input_data.count(')')
-    down = input_data.count('(')
-    floor = abs(up - down)
+    floor = abs(directions.count(')') - directions.count('('))
 
     assert floor == 138
+
+def test_part02():
+    directions = read_input("day01-input.txt")
+
+    position = 0
+    floor = 0
+    for i, f in enumerate(directions):
+        if f == '(': floor += 1
+        else: floor -= 1
+
+        if floor == -1:
+            position = i + 1
+            break
+
+    assert position == 1771