Fixup columns

panodata · Jun 16, 2020 · d2ebabb · d2ebabb
1 parent e93d814
commit d2ebabb
Showing 1 changed file with 34 additions and 1 deletion.
diff --git a/tests/test_data_collection.py b/tests/test_data_collection.py
@@ -119,9 +119,42 @@ def test_collect_dwd_data_empty():
 
 
 @pytest.mark.remote
-def test_fetch_and_parse_dwd_data():
+def test_fetch_and_parse_dwd_data_vanilla_columns():
     """ Test for data collection with real data """
 
+    data = collect_dwd_data(
+        station_ids=[1048],
+        parameter=Parameter.CLIMATE_SUMMARY,
+        time_resolution=TimeResolution.DAILY,
+        period_type=PeriodType.RECENT
+    )
+
+    assert list(data.columns.values) == [
+        'STATION_ID',
+        'DATE',
+        'QN_3',
+        'FX',
+        'FM',
+        'QN_4',
+        'RSK',
+        'RSKF',
+        'SDK',
+        'SHK_TAG',
+        'NM',
+        'VPM',
+        'PM',
+        'TMK',
+        'UPM',
+        'TXK',
+        'TNK',
+        'TGK',
+    ]
+
+
+@pytest.mark.remote
+def test_fetch_and_parse_dwd_data_humanized_columns():
+    """ Test for data collection with real data and humanized column names """
+
     data = collect_dwd_data(
         station_ids=[1048],
         parameter=Parameter.CLIMATE_SUMMARY,