black

Author: heidi-holm-4ss

fourinsight/engineroom/utils/_datamanage.py

@@ -635,7 +635,11 @@ def get_fun(ts_id, start, end, **kwargs):
                 aggregation_period = kwargs.pop("aggregation_period", "tick")
                 aggregation_function = kwargs.pop("aggregation_function", "mean")
                 return self._drio_client.get_samples_aggregate(
-                    ts_id, start=start, end=end, aggregation_period=aggregation_period, aggregation_function=aggregation_function
+                    ts_id,
+                    start=start,
+                    end=end,
+                    aggregation_period=aggregation_period,
+                    aggregation_function=aggregation_function,
                 )
 
         elif self.storage == "archive":

tests/test_datamanage.py

@@ -1119,7 +1119,7 @@ def test__get(self):
                 ),
             ]
         )
-    
+
     def test__get_warm_storage(self):
         # Mock the drio_client
         drio_client = Mock()
@@ -1139,9 +1139,9 @@ def test__get_warm_storage(self):
             drio_client,
             labels,
             storage="warm",
-            convert_date=True,   
-            raise_empty=False,   
-            aggregation_period="1h",   
+            convert_date=True,
+            raise_empty=False,
+            aggregation_period="1h",
             aggregation_function="min",
         )
 
@@ -1168,21 +1168,21 @@ def test__get_warm_storage(self):
                     start="<start-time>",
                     end="<end-time>",
                     aggregation_period="1h",
-                    aggregation_function="min"
+                    aggregation_function="min",
                 ),
                 call(
                     "timeseriesid-b",
                     start="<start-time>",
                     end="<end-time>",
                     aggregation_period="1h",
-                    aggregation_function="min"
+                    aggregation_function="min",
                 ),
                 call(
                     "timeseriesid-c",
                     start="<start-time>",
                     end="<end-time>",
                     aggregation_period="1h",
-                    aggregation_function="min"
+                    aggregation_function="min",
                 ),
             ]
         )
@@ -1201,11 +1201,7 @@ def test__get_warm_defaults(self):
 
         # Instantiate DrioDataSource with storage="warm", no overrides
         source = DrioDataSource(
-            drio_client,
-            labels,
-            storage="warm",
-            convert_date=True,
-            raise_empty=False
+            drio_client, labels, storage="warm", convert_date=True, raise_empty=False
         )
 
         # Call _get with defaults
@@ -1227,20 +1223,18 @@ def test__get_warm_defaults(self):
                     start="<start>",
                     end="<end>",
                     aggregation_period="tick",
-                    aggregation_function="mean"
+                    aggregation_function="mean",
                 ),
                 call(
                     "ts-y",
                     start="<start>",
                     end="<end>",
                     aggregation_period="tick",
-                    aggregation_function="mean"
+                    aggregation_function="mean",
                 ),
             ]
         )
 
-    
-
 
 class Test_NullDataSource:
     def test__init__(self):