policyengine 3.1.13__tar.gz → 3.1.14__tar.gz

{policyengine-3.1.13 → policyengine-3.1.14}/CHANGELOG.md

@@ -5,6 +5,12 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/), 
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [3.1.14] - 2025-12-10 21:59:24
+
+### Fixed
+
+- Improvements to loading taxbenefitsystems.
+
 ## [3.1.13] - 2025-12-10 19:29:28
 
 ### Fixed
@@ -275,6 +281,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 
 
+[3.1.14]: https://github.com/PolicyEngine/policyengine.py/compare/3.1.13...3.1.14
 [3.1.13]: https://github.com/PolicyEngine/policyengine.py/compare/3.1.12...3.1.13
 [3.1.12]: https://github.com/PolicyEngine/policyengine.py/compare/3.1.11...3.1.12
 [3.1.11]: https://github.com/PolicyEngine/policyengine.py/compare/3.1.10...3.1.11

{policyengine-3.1.13 → policyengine-3.1.14}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: policyengine
-Version: 3.1.13
+Version: 3.1.14
 Summary: A package to conduct policy analysis using PolicyEngine tax-benefit models.
 Author-email: PolicyEngine <hello@policyengine.org>
 License:                     GNU AFFERO GENERAL PUBLIC LICENSE

{policyengine-3.1.13 → policyengine-3.1.14}/changelog.yaml

@@ -225,3 +225,8 @@
     fixed:
     - Naming improvements.
   date: 2025-12-10 19:29:28
+- bump: patch
+  changes:
+    fixed:
+    - Improvements to loading taxbenefitsystems.
+  date: 2025-12-10 21:59:24

{policyengine-3.1.13 → policyengine-3.1.14}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "policyengine"
-version = "3.1.13"
+version = "3.1.14"
 description = "A package to conduct policy analysis using PolicyEngine tax-benefit models."
 readme = "README.md"
 authors = [

{policyengine-3.1.13 → policyengine-3.1.14}/src/policyengine/tax_benefit_models/uk/datasets.py

@@ -196,8 +196,8 @@ def load_datasets(
         for year in years:
             filepath = f"{data_folder}/{Path(dataset).stem}_year_{year}.h5"
             uk_dataset = PolicyEngineUKDataset(
-                name=f"{dataset}-year-{year}",
-                description=f"UK Dataset for year {year} based on {dataset}",
+                name=f"{Path(dataset).stem}-year-{year}",
+                description=f"UK Dataset for year {year} based on {Path(dataset).stem}",
                 filepath=filepath,
                 year=int(year),
             )

{policyengine-3.1.13 → policyengine-3.1.14}/src/policyengine/tax_benefit_models/us/datasets.py

@@ -317,8 +317,8 @@ def load_datasets(
         for year in years:
             filepath = f"{data_folder}/{Path(dataset).stem}_year_{year}.h5"
             us_dataset = PolicyEngineUSDataset(
-                name=f"{dataset}-year-{year}",
-                description=f"US Dataset for year {year} based on {dataset}",
+                name=f"{Path(dataset).stem}-year-{year}",
+                description=f"US Dataset for year {year} based on {Path(dataset).stem}",
                 filepath=filepath,
                 year=year,
             )

{policyengine-3.1.13 → policyengine-3.1.14}/src/policyengine/tax_benefit_models/us/model.py

@@ -141,6 +141,7 @@ class PolicyEngineUSLatest(TaxBenefitModelVersion):
                 parameter = Parameter(
                     id=self.id + "-" + param_node.name,
                     name=param_node.name,
+                    label=param_node.metadata.get("label"),
                     tax_benefit_model_version=self,
                     description=param_node.description,
                     data_type=type(param_node(2025)),

{policyengine-3.1.13 → policyengine-3.1.14}/src/policyengine.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: policyengine
-Version: 3.1.13
+Version: 3.1.14
 Summary: A package to conduct policy analysis using PolicyEngine tax-benefit models.
 Author-email: PolicyEngine <hello@policyengine.org>
 License:                     GNU AFFERO GENERAL PUBLIC LICENSE

{policyengine-3.1.13 → policyengine-3.1.14}/src/policyengine.egg-info/SOURCES.txt

@@ -79,4 +79,5 @@ src/policyengine/utils/plotting.py
 tests/test_aggregate.py
 tests/test_cache.py
 tests/test_change_aggregate.py
-tests/test_entity_mapping.py
+tests/test_entity_mapping.py
+tests/test_models.py

policyengine-3.1.14/tests/test_models.py

@@ -0,0 +1,68 @@
+"""Tests for UK and US tax-benefit model versions."""
+
+import pytest
+
+from policyengine.tax_benefit_models.uk import uk_latest
+from policyengine.tax_benefit_models.us import us_latest
+
+
+class TestUKModel:
+    """Tests for PolicyEngine UK model."""
+
+    def test_has_hundreds_of_parameters(self):
+        """UK model should have hundreds of parameters."""
+        assert len(uk_latest.parameters) >= 100
+
+    def test_has_hundreds_of_variables(self):
+        """UK model should have hundreds of variables."""
+        assert len(uk_latest.variables) >= 100
+
+    def test_parameters_have_values(self):
+        """Each parameter should have at least one parameter value."""
+        total_values = 0
+        for param in uk_latest.parameters[:100]:  # Check first 100 for speed
+            values = param.parameter_values
+            assert len(values) >= 1, f"Parameter {param.name} has no values"
+            total_values += len(values)
+
+        # Should have many parameter values in total
+        assert total_values >= 100
+
+    def test_parameter_values_have_required_fields(self):
+        """Parameter values should have start_date and value."""
+        for param in uk_latest.parameters[:50]:
+            for pv in param.parameter_values:
+                assert pv.start_date is not None
+                assert pv.value is not None
+                assert pv.parameter is param
+
+
+class TestUSModel:
+    """Tests for PolicyEngine US model."""
+
+    def test_has_hundreds_of_parameters(self):
+        """US model should have hundreds of parameters."""
+        assert len(us_latest.parameters) >= 100
+
+    def test_has_hundreds_of_variables(self):
+        """US model should have hundreds of variables."""
+        assert len(us_latest.variables) >= 100
+
+    def test_parameters_have_values(self):
+        """Each parameter should have at least one parameter value."""
+        total_values = 0
+        for param in us_latest.parameters[:100]:  # Check first 100 for speed
+            values = param.parameter_values
+            assert len(values) >= 1, f"Parameter {param.name} has no values"
+            total_values += len(values)
+
+        # Should have many parameter values in total
+        assert total_values >= 100
+
+    def test_parameter_values_have_required_fields(self):
+        """Parameter values should have start_date and value."""
+        for param in us_latest.parameters[:50]:
+            for pv in param.parameter_values:
+                assert pv.start_date is not None
+                assert pv.value is not None
+                assert pv.parameter is param

{policyengine-3.1.13 → policyengine-3.1.14}/uv.lock

@@ -1080,12 +1080,13 @@ wheels = [
 
 [[package]]
 name = "policyengine"
-version = "3.1.7"
+version = "3.1.12"
 source = { editable = "." }
 dependencies = [
     { name = "microdf-python" },
     { name = "pandas" },
     { name = "plotly" },
+    { name = "psutil" },
     { name = "pydantic" },
     { name = "requests" },
 ]
@@ -1133,6 +1134,7 @@ requires-dist = [
     { name = "policyengine-uk", marker = "extra == 'uk'", specifier = ">=2.51.0" },
     { name = "policyengine-us", marker = "extra == 'dev'", specifier = ">=1.213.1" },
     { name = "policyengine-us", marker = "extra == 'us'", specifier = ">=1.213.1" },
+    { name = "psutil", specifier = ">=5.9.0" },
     { name = "pydantic", specifier = ">=2.0.0" },
     { name = "pytest", marker = "extra == 'dev'" },
     { name = "pytest-asyncio", marker = "extra == 'dev'", specifier = ">=0.26.0" },