test: Re-add extension tests, which now pass

changelog_entry.yaml

@@ -1,4 +1,6 @@
 - bump: minor
   changes:
     changed:
-    - Shallow copy GroupEntities and PopulationEntity when cloning TaxBenefitSystem object
+    - Shallow copy GroupEntities and PopulationEntity when cloning TaxBenefitSystem object
+    added:
+    - Two tests related to extensions that were previously removed

tests/core/test_yaml.py

@@ -76,6 +76,13 @@ def test_with_reform(tax_benefit_system):
     )
 
 
+def test_with_extension(tax_benefit_system):
+    assert (
+        run_yaml_test(tax_benefit_system, "test_with_extension.yaml")
+        == EXIT_OK
+    )
+
+
 def test_with_anchors(tax_benefit_system):
     assert (
         run_yaml_test(tax_benefit_system, "test_with_anchors.yaml") == EXIT_OK

tests/fixtures/yaml_tests/test_with_extension.yaml

@@ -0,0 +1,28 @@
+- name: "Test using an extension"
+  period: 2017-01
+  extensions:
+    - policyengine_core.extension_template
+  input:
+    persons:
+      parent: {}
+      child1: {}
+    household:
+      parents: [parent]
+      children: [child1]
+  output:
+    local_town_child_allowance: 100
+
+- name: "Test using an extension"
+  period: 2017-01
+  extensions:
+    - policyengine_core.extension_template
+  input:
+    persons:
+      parent: {}
+      child1: {}
+      child2: {}
+    household:
+      parents: [parent]
+      children: [child1, child2]
+  output:
+    local_town_child_allowance: 200