Cleanup tests for cli and restapi

earthobservations · Oct 21, 2023 · 473b469 · 473b469
1 parent 12156af
commit 473b469
Show file tree

Hide file tree

Showing 3 changed files with 145 additions and 109 deletions.
diff --git a/tests/core/timeseries/test_io.py b/tests/core/timeseries/test_io.py
@@ -6,8 +6,6 @@
 import sqlite3
 from unittest import mock
 
-import duckdb
-import openpyxl
 import polars as pl
 import pytest
 from surrogate import surrogate
@@ -248,6 +246,8 @@ def test_export_unknown(default_settings):
 @pytest.mark.remote
 def test_export_spreadsheet(tmp_path, settings_si_false_wide_shape):
     """Test export of DataFrame to spreadsheet"""
+    import openpyxl
+
     # 1. Request data and save to .xlsx file.
     request = DwdObservationRequest(
         parameter=DwdObservationDataset.CLIMATE_SUMMARY,
@@ -598,6 +598,8 @@ def test_export_cratedb(
 @pytest.mark.remote
 def test_export_duckdb(settings_si_false, tmp_path):
     """Test export of DataFrame to duckdb"""
+    import duckdb
+
     request = DwdObservationRequest(
         parameter=DwdObservationDataset.CLIMATE_SUMMARY,
         resolution=DwdObservationResolution.DAILY,

diff --git a/tests/ui/test_cli.py b/tests/ui/test_cli.py
@@ -8,7 +8,7 @@
 import polars as pl
 import pytest
 from click.testing import CliRunner
-from dirty_equals import IsDict, IsFloat, IsStr
+from dirty_equals import IsFloat, IsStr
 
 from tests.conftest import IS_WINDOWS
 from wetterdienst.ui.cli import cli
@@ -346,16 +346,14 @@ def test_cli_values_json_multiple_datasets(capsys, caplog):
     )
     response = json.loads(result.stdout)
     item = response[12]
-    assert item == IsDict(
-        {
-            "station_id": "01048",
-            "dataset": "climate_summary",
-            "parameter": "wind_gust_max",
-            "date": "2020-06-30T00:00:00+00:00",
-            "value": 15.3,
-            "quality": 10.0,
-        }
-    )
+    assert item == {
+        "station_id": "01048",
+        "dataset": "climate_summary",
+        "parameter": "wind_gust_max",
+        "date": "2020-06-30T00:00:00+00:00",
+        "value": 15.3,
+        "quality": 10.0,
+    }
 
 
 @pytest.mark.remote