update test to use pkg resource file (#94)

IMMM-SFA · Sep 23, 2022 · f1b0f22 · f1b0f22
1 parent a8897ea
commit f1b0f22
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/mosartwmpy/_version.py b/mosartwmpy/_version.py
@@ -1 +1 @@
-__version__ = "0.4.0"
+__version__ = "0.4.1"
diff --git a/mosartwmpy/tests/test_calculate_water_constraints_by_farm.py b/mosartwmpy/tests/test_calculate_water_constraints_by_farm.py
@@ -14,6 +14,7 @@ class CalculateWaterConstraintsByFarmTest(unittest.TestCase):
     RUNOFF_FILE = pkg_resources.resource_filename('mosartwmpy', 'tests/runoff_1981_01_01.nc')
     DEMAND_FILE = pkg_resources.resource_filename('mosartwmpy', 'tests/demand_1981_01_01.nc')
     RESERVOIRS_FILE = pkg_resources.resource_filename('mosartwmpy', 'tests/reservoirs.nc')
+    CONSTRAINTS_FILE = pkg_resources.resource_filename('mosartwmpy', 'tests/test_land_water_constraints_by_farm.parquet')
 
     def test_calculate_water_constraints_by_farm(self):
         model = Model()
@@ -24,6 +25,7 @@ def test_calculate_water_constraints_by_farm(self):
         model.config['runoff.path'] = self.RUNOFF_FILE
         model.config['water_management.demand.path'] = self.DEMAND_FILE
         model.config['water_management.reservoirs.path'] = self.RESERVOIRS_FILE
+        model.config['water_management.demand.farmer_abm.land_water_constraints.path'] = self.CONSTRAINTS_FILE
 
         farmerABM = FarmerABM(model)