Add tests for strict dunder typing mode

Author: jairov4

mypyc/build.py

@@ -523,7 +523,7 @@ def mypycify(
         separate=separate is not False,
         target_dir=target_dir,
         include_runtime_files=include_runtime_files,
-        strict_dunders_typing=strict_dunder_typing,
+        strict_dunder_typing=strict_dunder_typing,
     )
 
     # Generate all the actual important C code

mypyc/options.py

@@ -14,7 +14,7 @@ def __init__(
         include_runtime_files: bool | None = None,
         capi_version: tuple[int, int] | None = None,
         python_version: tuple[int, int] | None = None,
-        strict_dunders_typing: bool = False,
+        strict_dunder_typing: bool = False,
     ) -> None:
         self.strip_asserts = strip_asserts
         self.multi_file = multi_file
@@ -37,4 +37,4 @@ def __init__(
         # By enabling this option, this convention is no longer valid and the dunder
         # will assume the return type of the method strictly, which can lead to
         # more optimization opportunities.
-        self.strict_dunders_typing = strict_dunders_typing
+        self.strict_dunders_typing = strict_dunder_typing

mypyc/test/test_run.py

@@ -140,6 +140,7 @@ class TestRun(MypycDataSuite):
     optional_out = True
     multi_file = False
     separate = False  # If True, using separate (incremental) compilation
+    strict_dunder_typing = False
 
     def run_case(self, testcase: DataDrivenTestCase) -> None:
         # setup.py wants to be run from the root directory of the package, which we accommodate
@@ -232,7 +233,11 @@ def run_case_step(self, testcase: DataDrivenTestCase, incremental_step: int) ->
         groups = construct_groups(sources, separate, len(module_names) > 1)
 
         try:
-            compiler_options = CompilerOptions(multi_file=self.multi_file, separate=self.separate)
+            compiler_options = CompilerOptions(
+                multi_file=self.multi_file,
+                separate=self.separate,
+                strict_dunder_typing=self.strict_dunder_typing,
+            )
             result = emitmodule.parse_and_typecheck(
                 sources=sources,
                 options=options,
@@ -401,6 +406,14 @@ class TestRunSeparate(TestRun):
     files = ["run-multimodule.test", "run-mypy-sim.test"]
 
 
+class TestRunStrictDunderTyping(TestRun):
+    """Run the tests with strict dunder typing."""
+
+    strict_dunder_typing = True
+    test_name_suffix = "_dunder_typing"
+    files = ["run-dunders.test", "run-floats.test"]
+
+
 def fix_native_line_number(message: str, fnam: str, delta: int) -> str:
     """Update code locations in test case output to point to the .test file.