mypy: stricly type tests

inducer · Jan 10, 2025 · 8f70149 · 8f70149
1 parent afec7a2
commit 8f70149
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/test/test_identical_symbols.py b/test/test_identical_symbols.py
@@ -1,7 +1,9 @@
+from types import ModuleType
+
 import pytest
 
 
-def make_greet_mod(greeting):
+def make_greet_mod(greeting: str) -> ModuleType:
     from cgen import (
         Block,
         Const,
@@ -19,7 +21,7 @@ def make_greet_mod(greeting):
     mod.add_function(
             FunctionBody(
                 FunctionDeclaration(Const(Pointer(Value("char", "greet"))), []),
-                Block([Statement('return "%s"' % greeting)])
+                Block([Statement(f'return "{greeting}"')])
                 ))
 
     from codepy.toolchain import guess_toolchain
@@ -29,7 +31,7 @@ def make_greet_mod(greeting):
 @pytest.mark.xfail(reason="You probably don't have "
         "Boost.Python installed where I am looking for it, "
         "and that's OK.")
-def test_identical_symbols():
+def test_identical_symbols() -> None:
     us = make_greet_mod("Hi there")
     aussie = make_greet_mod("G'day")
 

diff --git a/test/test_two_stage_compile.py b/test/test_two_stage_compile.py
@@ -4,7 +4,7 @@
 from codepy.toolchain import guess_toolchain
 
 
-def test():
+def test() -> None:
     toolchain = guess_toolchain()
 
     module_code = """