feat: Specify the point source directory using an environment variable (#31)

The `SPAEMIS_POINT_SOURCE_DIRECTORY` is used

Author: lewisjared

data/raw/configuration/scenarios/ssp245-high-production_victoria.yaml

@@ -30,7 +30,7 @@ scalers:
       allow_missing: true
       method:
         name: point_source
-        point_sources: point_sources/hysupply_locations.csv
+        point_sources: hysupply_locations.csv
         source_timeseries: high_production
         source_filters:
           - product: H2
@@ -39,7 +39,7 @@ scalers:
       allow_missing: true
       method:
         name: point_source
-        point_sources: point_sources/hysupply_locations.csv
+        point_sources: hysupply_locations.csv
         source_timeseries: high_production
         source_filters:
           - product: NOx

src/spaemis/scaling/point_source.py

@@ -114,12 +114,22 @@ def __call__(
     def create_from_config(cls, method: ScalerMethod) -> "PointSourceScaler":
         """
         Create a scaler from configuration
+
+        Point source information will be loaded from a directory specified by the
+        environment variable `SPAEMIS_POINT_SOURCE_DIRECTORY`. This defaults to
+        `data/raw/configuration/point_sources` if no environment variable
+        is specified.
         """
         if not isinstance(method, PointSourceMethod):
             raise TypeError("Incompatible configuration")
 
+        point_source_directory = os.environ.get(
+            "SPAEMIS_POINT_SOURCE_DIRECTORY",
+            os.path.join(RAW_DATA_DIR, "configuration", "point_sources"),
+        )
+
         point_info = pd.read_csv(
-            os.path.join(RAW_DATA_DIR, "configuration", method.point_sources)
+            os.path.join(point_source_directory, method.point_sources)
         )
         points = [
             Point(lat=item["lat"], lon=item["lon"])

tests/unit/scaling/test_scaling.py

@@ -275,7 +275,7 @@ def test_run(self, inventory, caplog):
 
         scaler = PointSourceScaler.create_from_config(
             PointSourceMethod(
-                point_sources="point_sources/hysupply_locations.csv",
+                point_sources="hysupply_locations.csv",
                 source_timeseries="high_production",
                 source_filters=[{"product": "H2"}],
             )