Linting.

api/load-test/locust-ewc.py

@@ -22,7 +22,10 @@ def on_start(self):
         stations = response.json()["features"]
         self.stations = {s["id"]: s["properties"]["parameter-name"] for s in stations}
         self.station_ids = list(self.stations.keys())
-        self.stations_by_location = {(s["geometry"]["coordinates"][0],s["geometry"]["coordinates"][1]): s["properties"]["parameter-name"] for s in stations}
+        self.stations_by_location = {
+            (s["geometry"]["coordinates"][0], s["geometry"]["coordinates"][1]): s["properties"]["parameter-name"]
+            for s in stations
+        }
         self.station_locations = list(self.stations_by_location.keys())
 
     # @task
@@ -39,13 +42,13 @@ def on_start(self):
     def get_data_single_station_single_parameter_last_x_hours(self):
         hours = random.choice(hours_choice)
         date_time = datetime.now(UTC) - timedelta(hours=hours)
-        dt_string = date_time.strftime('%Y-%m-%dT%H:%M:%SZ')
+        dt_string = date_time.strftime("%Y-%m-%dT%H:%M:%SZ")
         station_id = random.choice(self.station_ids)
         parameter = random.choice(self.stations[station_id])
         response = self.client.get(
             f"/collections/observations/locations/{station_id}?parameter-name={parameter}&datetime={dt_string}/..",
             name=f"location {hours:02d} hours",
-            headers=headers
+            headers=headers,
         )
 
     @task
@@ -55,27 +58,23 @@ def get_data_single_position_single_parameter(self):
         response = self.client.get(
             f"/collections/observations/position?coords=POINT({lon} {lat})&parameter-name={parameter}",
             name="position",
-            headers=headers
+            headers=headers,
         )
 
     @task
     def get_data_area_single_parameter_last_hour(self):
         date_time = datetime.now(UTC) - timedelta(hours=1)
-        dt_string = date_time.strftime('%Y-%m-%dT%H:%M:%SZ')
+        dt_string = date_time.strftime("%Y-%m-%dT%H:%M:%SZ")
         standard_name = random.choice(common_standard_names)
         (cx, cy) = random.choice(self.station_locations)
-        sz = random.choice(polygon_size)/2.0
+        sz = random.choice(polygon_size) / 2.0
         left = cx - sz
         bottom = cy - sz
         right = cx + sz
         top = cy + sz
         polygon = f"POLYGON(({left} {bottom},{right} {bottom},{right} {top},{left} {top},{left} {bottom}))"
         url = f"/collections/observations/area?coords={polygon}&standard_names={standard_name}&datetime={dt_string}/.."
-        response = self.client.get(
-            url,
-            name=f"area {sz*2.0}deg x {sz*2.0}deg x 1h",
-            headers=headers
-        )
+        response = self.client.get(url, name=f"area {sz*2.0}deg x {sz*2.0}deg x 1h", headers=headers)
         # if sz == 2.0:
         #     j = response.json()
         #     # print(sz*2.0)

api/load-test/locustfile.py

@@ -3,8 +3,12 @@
 from locust import HttpUser
 from locust import task
 
-parameters = ["wind_speed:10.0:mean:PT10M", "wind_speed:10.0:mean:PT10M", "relative_humidity:2.0:mean:PT1M",
-              "air_temperature:2.0:mean:PT1M"]
+parameters = [
+    "wind_speed:10.0:mean:PT10M",
+    "wind_speed:10.0:mean:PT10M",
+    "relative_humidity:2.0:mean:PT1M",
+    "air_temperature:2.0:mean:PT1M",
+]
 # fmt: off
 stations = [
     "0-20000-0-06203", "0-20000-0-06204", "0-20000-0-06205", "0-20000-0-06207", "0-20000-0-06208", "0-20000-0-06211",