Canada district structure

policyengine_canada/variables/household/demographic/geographic/district/district.py

@@ -0,0 +1,15 @@
+from policyengine_canada.model_api import *
+from policyengine_canada.variables.household.demographic.geographic.district.district_enum import (
+    District,
+)
+
+import numpy as np
+
+
+class county(Variable):
+    value_type = Enum
+    possible_values = District
+    default_value = District.UNKNOWN
+    entity = Household
+    label = "District"
+    definition_period = ETERNITY

policyengine_canada/variables/household/demographic/geographic/district/district_enum.py

@@ -0,0 +1,5 @@
+from policyengine_canada.model_api import *
+
+
+class District(Enum):
+    UNKNOWN = "Unknown"

policyengine_canada/variables/household/demographic/geographic/district/district_str.py

@@ -0,0 +1,12 @@
+from policyengine_canada.model_api import *
+
+
+class district_str(Variable):
+    value_type = str
+    entity = Household
+    label = "Dsitrict (string)"
+    documentation = "District variable, stored as a string"
+    definition_period = ETERNITY
+
+    def formula(household, period, parameters):
+        return household("district", period).decode_to_str()