Add type annotations (#141)

Author: mandel

tests/test_examples_run.py

@@ -4,6 +4,8 @@
 from dataclasses import dataclass
 from typing import Optional
 
+from pytest import CaptureFixture, MonkeyPatch
+
 from pdl import pdl
 from pdl.pdl_ast import ScopeType
 from pdl.pdl_interpreter import PDLRuntimeError
@@ -144,7 +146,7 @@ class InputsType:
 ]
 
 
-def test_valid_programs(capsys, monkeypatch) -> None:
+def test_valid_programs(capsys: CaptureFixture[str], monkeypatch: MonkeyPatch) -> None:
     actual_parse_error: set[str] = set()
     actual_runtime_error: set[str] = set()
     wrong_results = {}

tests/test_line_table.py

@@ -1,3 +1,5 @@
+from pytest import CaptureFixture
+
 from pdl.pdl_interpreter import generate
 
 
@@ -21,7 +23,7 @@ def do_test(t, capsys):
 }
 
 
-def test_line(capsys):
+def test_line(capsys: CaptureFixture[str]):
     do_test(line, capsys)
 
 
@@ -35,7 +37,7 @@ def test_line(capsys):
 }
 
 
-def test_line1(capsys):
+def test_line1(capsys: CaptureFixture[str]):
     do_test(line1, capsys)
 
 
@@ -49,7 +51,7 @@ def test_line1(capsys):
 }
 
 
-def test_line3(capsys):
+def test_line3(capsys: CaptureFixture[str]):
     do_test(line3, capsys)
 
 
@@ -63,7 +65,7 @@ def test_line3(capsys):
 }
 
 
-def test_line4(capsys):
+def test_line4(capsys: CaptureFixture[str]):
     do_test(line4, capsys)
 
 
@@ -77,7 +79,7 @@ def test_line4(capsys):
 }
 
 
-def test_line7(capsys):
+def test_line7(capsys: CaptureFixture[str]):
     do_test(line7, capsys)
 
 
@@ -91,7 +93,7 @@ def test_line7(capsys):
 }
 
 
-def test_line8(capsys):
+def test_line8(capsys: CaptureFixture[str]):
     do_test(line8, capsys)
 
 
@@ -105,7 +107,7 @@ def test_line8(capsys):
 }
 
 
-def test_line9(capsys):
+def test_line9(capsys: CaptureFixture[str]):
     do_test(line9, capsys)
 
 
@@ -118,7 +120,7 @@ def test_line9(capsys):
 }
 
 
-def test_line10(capsys):
+def test_line10(capsys: CaptureFixture[str]):
     do_test(line10, capsys)
 
 
@@ -131,7 +133,7 @@ def test_line10(capsys):
 }
 
 
-def test_line11(capsys):
+def test_line11(capsys: CaptureFixture[str]):
     do_test(line11, capsys)
 
 
@@ -145,7 +147,7 @@ def test_line11(capsys):
 }
 
 
-def test_line12(capsys):
+def test_line12(capsys: CaptureFixture[str]):
     do_test(line12, capsys)
 
 
@@ -159,7 +161,7 @@ def test_line12(capsys):
 }
 
 
-def test_line13(capsys):
+def test_line13(capsys: CaptureFixture[str]):
     do_test(line13, capsys)
 
 
@@ -173,7 +175,7 @@ def test_line13(capsys):
 }
 
 
-def test_line14(capsys):
+def test_line14(capsys: CaptureFixture[str]):
     do_test(line14, capsys)
 
 
@@ -186,7 +188,7 @@ def test_line14(capsys):
 }
 
 
-def test_line15(capsys):
+def test_line15(capsys: CaptureFixture[str]):
     do_test(line15, capsys)
 
 
@@ -200,7 +202,7 @@ def test_line15(capsys):
 }
 
 
-def test_line16(capsys):
+def test_line16(capsys: CaptureFixture[str]):
     do_test(line16, capsys)
 
 
@@ -214,7 +216,7 @@ def test_line16(capsys):
 }
 
 
-def test_line17(capsys):
+def test_line17(capsys: CaptureFixture[str]):
     do_test(line17, capsys)
 
 
@@ -227,7 +229,7 @@ def test_line17(capsys):
 }
 
 
-def test_line18(capsys):
+def test_line18(capsys: CaptureFixture[str]):
     do_test(line18, capsys)
 
 
@@ -242,7 +244,7 @@ def test_line18(capsys):
 }
 
 
-def test_line19(capsys):
+def test_line19(capsys: CaptureFixture[str]):
     do_test(line19, capsys)
 
 
@@ -255,7 +257,7 @@ def test_line19(capsys):
 }
 
 
-def test_line20(capsys):
+def test_line20(capsys: CaptureFixture[str]):
     do_test(line20, capsys)
 
 
@@ -268,7 +270,7 @@ def test_line20(capsys):
 }
 
 
-def test_line21(capsys):
+def test_line21(capsys: CaptureFixture[str]):
     do_test(line21, capsys)
 
 
@@ -281,7 +283,7 @@ def test_line21(capsys):
 }
 
 
-def test_line22(capsys):
+def test_line22(capsys: CaptureFixture[str]):
     do_test(line22, capsys)
 
 
@@ -294,7 +296,7 @@ def test_line22(capsys):
 }
 
 
-def test_line23(capsys):
+def test_line23(capsys: CaptureFixture[str]):
     do_test(line23, capsys)
 
 
@@ -307,7 +309,7 @@ def test_line23(capsys):
 }
 
 
-def test_line24(capsys):
+def test_line24(capsys: CaptureFixture[str]):
     do_test(line24, capsys)
 
 
@@ -335,7 +337,7 @@ def test_line24(capsys):
 }
 
 
-def test_line26(capsys):
+def test_line26(capsys: CaptureFixture[str]):
     do_test(line26, capsys)
 
 
@@ -348,7 +350,7 @@ def test_line26(capsys):
 }
 
 
-def test_line27(capsys):
+def test_line27(capsys: CaptureFixture[str]):
     do_test(line27, capsys)
 
 
@@ -361,7 +363,7 @@ def test_line27(capsys):
 }
 
 
-def test_line28(capsys):
+def test_line28(capsys: CaptureFixture[str]):
     do_test(line28, capsys)
 
 
@@ -374,7 +376,7 @@ def test_line28(capsys):
 }
 
 
-def test_line29(capsys):
+def test_line29(capsys: CaptureFixture[str]):
     do_test(line29, capsys)
 
 
@@ -387,7 +389,7 @@ def test_line29(capsys):
 }
 
 
-def test_line30(capsys):
+def test_line30(capsys: CaptureFixture[str]):
     do_test(line30, capsys)
 
 
@@ -402,5 +404,5 @@ def test_line30(capsys):
 }
 
 
-def test_line31(capsys):
+def test_line31(capsys: CaptureFixture[str]):
     do_test(line31, capsys)