policyengine-us 1.389.0__py3-none-any.whl → 1.389.1__py3-none-any.whl

policyengine_us/tests/run_selective_tests.py

@@ -289,6 +289,11 @@ class SelectiveTestRunner:
 
             # Special handling for test files themselves
             if "tests" in file and file.endswith(".py"):
+                # Skip test infrastructure files that shouldn't trigger all tests
+                if file.endswith(
+                    ("run_selective_tests.py", "test_batched.py")
+                ):
+                    continue
                 # Add the directory containing the test file
                 test_dir = os.path.dirname(file)
                 if test_dir:
@@ -314,7 +319,13 @@ class SelectiveTestRunner:
 
         return existing_test_paths
 
-    def run_tests(self, test_paths: Set[str], verbose: bool = False) -> int:
+    def run_tests(
+        self,
+        test_paths: Set[str],
+        verbose: bool = False,
+        with_coverage: bool = False,
+        changed_files: Set[str] = None,
+    ) -> int:
         """Run pytest on specified test paths."""
         if not test_paths:
             print("No relevant tests found for changed files.")
@@ -325,7 +336,86 @@ class SelectiveTestRunner:
             print(f"  - {path}")
 
         # Construct pytest command
-        pytest_args = ["policyengine-core", "test", "-c", "policyengine_us"]
+        if with_coverage:
+            # Use coverage to run the tests
+            import sys
+
+            # First check if coverage is importable
+            try:
+                import coverage
+
+                print(f"Coverage version: {coverage.__version__}")
+            except ImportError as e:
+                print(f"ERROR: Coverage module not found: {e}")
+                print(
+                    "Please ensure coverage is installed: pip install coverage"
+                )
+                return 1
+
+            # Only track coverage for the specific files that changed in the PR
+            include_patterns = []
+            if changed_files:
+                # Only include Python files that were actually changed (excluding test directories)
+                changed_py_files = [
+                    f
+                    for f in changed_files
+                    if f.endswith(".py") and "/tests/" not in f
+                ]
+                if changed_py_files:
+                    include_patterns = changed_py_files
+            else:
+                # Fallback to directory-based patterns if no changed files provided
+                for test_path in test_paths:
+                    # Convert test path to variable path
+                    if "tests/policy/baseline/" in test_path:
+                        var_path = test_path.replace(
+                            "tests/policy/baseline/", "variables/"
+                        )
+                        include_patterns.append(f"{var_path}/**/*.py")
+                        include_patterns.append(f"{var_path}/*.py")
+                    elif "tests/policy/reform/" in test_path:
+                        include_patterns.append(
+                            "policyengine_us/reforms/**/*.py"
+                        )
+                        include_patterns.append("policyengine_us/reforms/*.py")
+                    elif "tests/policy/contrib/" in test_path:
+                        include_patterns.append(
+                            "policyengine_us/parameters/contrib/**/*.py"
+                        )
+                        include_patterns.append(
+                            "policyengine_us/parameters/contrib/*.py"
+                        )
+
+            pytest_args = [
+                sys.executable,
+                "-m",
+                "coverage",
+                "run",
+                "-a",
+                "--branch",
+            ]
+
+            # Add --include flag to only track relevant files
+            if include_patterns:
+                include_pattern = ",".join(include_patterns)
+                pytest_args.extend(["--include", include_pattern])
+
+            pytest_args.extend(
+                [
+                    "-m",
+                    "policyengine_core.scripts.policyengine_command",
+                    "test",
+                    "-c",
+                    "policyengine_us",
+                ]
+            )
+        else:
+            pytest_args = [
+                "policyengine-core",
+                "test",
+                "-c",
+                "policyengine_us",
+            ]
 
         # Add test paths
         pytest_args.extend(sorted(test_paths))
@@ -409,6 +499,11 @@ def main():
         action="store_true",
         help="Show debug information about git state",
     )
+    parser.add_argument(
+        "--coverage",
+        action="store_true",
+        help="Run tests with coverage measurement",
+    )
 
     args = parser.parse_args()
 
@@ -508,7 +603,14 @@ def main():
         print("Consider running all tests with --all flag.")
         sys.exit(0)
 
-    sys.exit(runner.run_tests(test_paths, verbose=args.verbose))
+    sys.exit(
+        runner.run_tests(
+            test_paths,
+            verbose=args.verbose,
+            with_coverage=args.coverage,
+            changed_files=changed_files if args.coverage else None,
+        )
+    )
 
 
 if __name__ == "__main__":

{policyengine_us-1.389.0.dist-info → policyengine_us-1.389.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: policyengine-us
-Version: 1.389.0
+Version: 1.389.1
 Summary: Add your description here.
 Author-email: PolicyEngine <hello@policyengine.org>
 License-File: LICENSE

{policyengine_us-1.389.0.dist-info → policyengine_us-1.389.1.dist-info}/RECORD

@@ -3190,7 +3190,7 @@ policyengine_us/reforms/tax_exempt/__init__.py,sha256=v3EbtUk6a2jsi9MrOlyb8ckzPy
 policyengine_us/reforms/tax_exempt/tax_exempt_reform.py,sha256=bkW91XMJ-jd23nhq4-PfWJv0YW2whVn98cQikbq6-Z8,2427
 policyengine_us/reforms/treasury/__init__.py,sha256=406jIbu32B57tUKhVLflWxlXf3S4SWjclVqlEtMeMwg,92
 policyengine_us/reforms/treasury/repeal_dependent_exemptions.py,sha256=-xyMCtUK_dIBn2aAmhPW8pJEZjxcgOZzWKEUSI3w5e8,1354
-policyengine_us/tests/run_selective_tests.py,sha256=AYvT1yM4bkOy4_2RtJEfpiV-l9lrlj86cl11zs8TsrE,18931
+policyengine_us/tests/run_selective_tests.py,sha256=kVuz9-CEXL2db8MuFi1J3cHj_Mqx28retvAPX8N3LCY,22571
 policyengine_us/tests/test_batched.py,sha256=K1LN3IIbWIioXWpFXZ6Hlch-BeglV3QPc1zxAbUeP-M,11491
 policyengine_us/tests/code_health/parameters.py,sha256=e9VKC8dmmB_dTafOO90WJuVu-PAogoLAaa5QZd2rY-s,1126
 policyengine_us/tests/code_health/variable_names.py,sha256=hY4ucqPwBD7v_fvnBpzexJDf0yCGpF4Sueff4z4rQ24,554
@@ -8378,8 +8378,8 @@ policyengine_us/variables/input/farm_income.py,sha256=BEKxYmHNNnWJAAvULl5qZJigy5
 policyengine_us/variables/input/geography.py,sha256=Ux0ueAf0rhZaflyEqz81UuXP3xKCKBDvoO3CrKhiQEc,5421
 policyengine_us/variables/input/self_employment_income.py,sha256=PwsGz8R4lRikKWUYOhsC0qosNNLXq4f5SQmfw4S3mk8,511
 policyengine_us/variables/input/self_employment_income_before_lsr.py,sha256=E8fcX9Nlyqz8dziHhQv_euutdmoIwFMMWePUwbbwv_w,379
-policyengine_us-1.389.0.dist-info/METADATA,sha256=oRKU3Ud2HNPdNXxoDeuit_z5MXjpbsEDe1prifcuT-c,1649
-policyengine_us-1.389.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-policyengine_us-1.389.0.dist-info/entry_points.txt,sha256=MLaqNyNTbReALyKNkde85VkuFFpdPWAcy8VRG1mjczc,57
-policyengine_us-1.389.0.dist-info/licenses/LICENSE,sha256=2N5ReRelkdqkR9a-KP-y-shmcD5P62XoYiG-miLTAzo,34519
-policyengine_us-1.389.0.dist-info/RECORD,,
+policyengine_us-1.389.1.dist-info/METADATA,sha256=-4H82MCggJ3clZoCBNAL5GXdJSmtr96COTxL1VzK1SU,1649
+policyengine_us-1.389.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+policyengine_us-1.389.1.dist-info/entry_points.txt,sha256=MLaqNyNTbReALyKNkde85VkuFFpdPWAcy8VRG1mjczc,57
+policyengine_us-1.389.1.dist-info/licenses/LICENSE,sha256=2N5ReRelkdqkR9a-KP-y-shmcD5P62XoYiG-miLTAzo,34519
+policyengine_us-1.389.1.dist-info/RECORD,,