linted

geomagio/api/ws/DataApiQuery.py

@@ -74,7 +74,9 @@ class DataApiQuery(BaseModel):
     data_host: Union[DataHost, str] = DataHost.DEFAULT
 
     @field_validator("starttime", mode="before")
-    def validate_starttime(cls, starttime: CustomUTCDateTimeType) -> CustomUTCDateTimeType:
+    def validate_starttime(
+        cls, starttime: CustomUTCDateTimeType
+    ) -> CustomUTCDateTimeType:
         if not starttime:
             return default_starttime()
         else:

test/DataApiQuery_test.py

@@ -29,6 +29,7 @@ def test_DataApiQuery_defaults():
     # assumes the env var DATA_HOST is not set
     assert_equal(query.data_host.value, "edgecwb.usgs.gov")
 
+
 def test_DataApiQuery_starttime_is_none():
     query = DataApiQuery(id="BOU", starttime=None)
 
@@ -46,6 +47,7 @@ def test_DataApiQuery_starttime_is_none():
     # assumes the env var DATA_HOST is not set
     assert_equal(query.data_host.value, "edgecwb.usgs.gov")
 
+
 def test_DataApiQuery_valid():
     query = DataApiQuery(
         id="DED",
@@ -125,6 +127,7 @@ def test_DataApiQuery_default_only_endtime():
     assert_equal(query.format, OutputFormat.IAGA2002)
     assert_equal(query.data_host, DataHost.DEFAULT)
 
+
 def test_DataApiQuery_default_only_invalid_endtime():
     # test an invalid endtime that is before default starttime
     query = None
@@ -133,7 +136,7 @@ def test_DataApiQuery_default_only_invalid_endtime():
     except Exception as e:
         err = e.errors()
         assert "Value error, Starttime must be before endtime." == err[0]["msg"]
-    
+
     assert_equal(query, None)
 
 

test/api_test/ws_test/data_test.py

@@ -39,11 +39,10 @@ def test_get_data_query(test_client):
     assert_equal(query.format, "iaga2002")
     assert_equal(query.data_type, "R1")
 
+
 def test_get_data_query_no_starttime(test_client):
     """test.api_test.ws_test.data_test.test_get_data_query_no_starttime()"""
-    response = test_client.get(
-        "/query/?id=BOU&elements=X,Y,Z,F"
-    )
+    response = test_client.get("/query/?id=BOU&elements=X,Y,Z,F")
     query = DataApiQuery(**response.json())
     assert_equal(query.id, "BOU")
 
@@ -55,6 +54,7 @@ def test_get_data_query_no_starttime(test_client):
     # expect endtime to default to 1 day after starttiime
     assert_equal(query.endtime, expected + (86400 - 0.001))
 
+
 async def test_get_data_query_extra_params(test_client):
     with pytest.raises(ValueError) as error:
         response = await test_client.get(

test/api_test/ws_test/filter_test.py

@@ -40,9 +40,7 @@ def test_get_filter_data_query(test_client):
 
 def test_get_filter_data_query_no_starttime(test_client):
     """test.api_test.ws_test.data_test.test_get_filter_data_query_no_starttime()"""
-    response = test_client.get(
-        "/filter/?id=BOU&elements=X,Y,Z,F"
-    )
+    response = test_client.get("/filter/?id=BOU&elements=X,Y,Z,F")
     query = FilterApiQuery(**response.json())
     assert_equal(query.id, "BOU")
 

test/pydantic_utcdatetime_test.py

@@ -16,6 +16,7 @@ def test_UTCDateTime_string():
 
     assert_equal(t.starttime, UTCDateTime(2024, 11, 5, 0, 0))
 
+
 def test_UTCDateTime_ISO860_string():
     # `ISO8601:2004`_ string from obspy docs
     t = TimeClass(starttime="2024-11-05T00:00:00.00")
@@ -48,6 +49,9 @@ def test_invalid():
     except Exception as e:
         err = e.errors()
         assert "Input should be an instance of UTCDateTime" == err[0]["msg"]
-        assert "Value error, Invalid time type. See obspy UTCDateTime for more information." == err[1]["msg"]
+        assert (
+            "Value error, Invalid time type. See obspy UTCDateTime for more information."
+            == err[1]["msg"]
+        )
 
     assert_equal(t, None)