Skip to content
Snippets Groups Projects
Commit dfbbdd31 authored by leufen1's avatar leufen1
Browse files

moved table test, /close #286

parent 69fb063b
No related branches found
No related tags found
6 merge requests!319add all changes of dev into release v1.4.0 branch,!318Resolve "release v1.4.0",!283Merge latest develop into falcos issue,!264Merge develop into felix_issue287_tech-wrf-datahandler-should-inherit-from-singlestationdatahandler,!259Draft: Resolve "WRF-Datahandler should inherit from SingleStationDatahandler",!257Resolve "store performance measures"
Pipeline #61730 passed
import pandas as pd
import numpy as np
from mlair.helpers import tables
class TestTables:
def test_create_column_format_for_tex(self):
df = pd.DataFrame(np.ones((2, 1)))
df_col = tables.create_column_format_for_tex(df) # len: 1+1
assert df_col == 'lr'
assert len(df_col) == 2
df = pd.DataFrame(np.ones((2, 2)))
df_col = tables.create_column_format_for_tex(df) # len: 2+1
assert df_col == 'lcr'
assert len(df_col) == 3
df = pd.DataFrame(np.ones((2, 3)))
df_col = tables.create_column_format_for_tex(df) # len: 3+1
assert df_col == 'lccr'
assert len(df_col) == 4
......@@ -140,40 +140,28 @@ class TestPreProcessing:
data_preparation = AbstractDataHandler
stations = ['DEBW107', 'DEBY081']
assert pre.transformation(data_preparation, stations) is None
class data_preparation_no_trans: pass
assert pre.transformation(data_preparation_no_trans, stations) is None
@pytest.fixture
def dummy_df(self):
data_dict = {'station_name': {'DEBW013': 'Stuttgart Bad Cannstatt', 'DEBW076': 'Baden-Baden',
'DEBW087': 'Schwäbische_Alb', 'DEBW107': 'Tübingen',
'DEBY081': 'Garmisch-Partenkirchen/Kreuzeckbahnstraße', '# Stations': np.nan,
'# Samples': np.nan},
'station_lon': {'DEBW013': 9.2297, 'DEBW076': 8.2202, 'DEBW087': 9.2076, 'DEBW107': 9.0512,
'DEBY081': 11.0631, '# Stations': np.nan, '# Samples': np.nan},
'station_lat': {'DEBW013': 48.8088, 'DEBW076': 48.7731, 'DEBW087': 48.3458, 'DEBW107': 48.5077,
'DEBY081': 47.4764, '# Stations': np.nan, '# Samples': np.nan},
'station_alt': {'DEBW013': 235.0, 'DEBW076': 148.0, 'DEBW087': 798.0, 'DEBW107': 325.0,
'DEBY081': 735.0, '# Stations': np.nan, '# Samples': np.nan},
'train': {'DEBW013': 1413, 'DEBW076': 3002, 'DEBW087': 3016, 'DEBW107': 1782, 'DEBY081': 2837,
'# Stations': 6, '# Samples': 12050},
'val': {'DEBW013': 698, 'DEBW076': 715, 'DEBW087': 700, 'DEBW107': 701, 'DEBY081': 456,
'# Stations': 6, '# Samples': 3270},
'test': {'DEBW013': 1066, 'DEBW076': 696, 'DEBW087': 1080, 'DEBW107': 1080, 'DEBY081': 700,
'# Stations': 6, '# Samples': 4622}}
df = pd.DataFrame.from_dict(data_dict)
return df
def test_create_column_format_for_tex(self):
df = pd.DataFrame(np.ones((2, 1)))
df_col = PreProcessing.create_column_format_for_tex(df) # len: 1+1
assert df_col == 'lr'
assert len(df_col) == 2
df = pd.DataFrame(np.ones((2, 2)))
df_col = PreProcessing.create_column_format_for_tex(df) # len: 2+1
assert df_col == 'lcr'
assert len(df_col) == 3
df = pd.DataFrame(np.ones((2, 3)))
df_col = PreProcessing.create_column_format_for_tex(df) # len: 3+1
assert df_col == 'lccr'
assert len(df_col) == 4
# @pytest.fixture
# def dummy_df(self):
# data_dict = {'station_name': {'DEBW013': 'Stuttgart Bad Cannstatt', 'DEBW076': 'Baden-Baden',
# 'DEBW087': 'Schwäbische_Alb', 'DEBW107': 'Tübingen',
# 'DEBY081': 'Garmisch-Partenkirchen/Kreuzeckbahnstraße', '# Stations': np.nan,
# '# Samples': np.nan},
# 'station_lon': {'DEBW013': 9.2297, 'DEBW076': 8.2202, 'DEBW087': 9.2076, 'DEBW107': 9.0512,
# 'DEBY081': 11.0631, '# Stations': np.nan, '# Samples': np.nan},
# 'station_lat': {'DEBW013': 48.8088, 'DEBW076': 48.7731, 'DEBW087': 48.3458, 'DEBW107': 48.5077,
# 'DEBY081': 47.4764, '# Stations': np.nan, '# Samples': np.nan},
# 'station_alt': {'DEBW013': 235.0, 'DEBW076': 148.0, 'DEBW087': 798.0, 'DEBW107': 325.0,
# 'DEBY081': 735.0, '# Stations': np.nan, '# Samples': np.nan},
# 'train': {'DEBW013': 1413, 'DEBW076': 3002, 'DEBW087': 3016, 'DEBW107': 1782, 'DEBY081': 2837,
# '# Stations': 6, '# Samples': 12050},
# 'val': {'DEBW013': 698, 'DEBW076': 715, 'DEBW087': 700, 'DEBW107': 701, 'DEBY081': 456,
# '# Stations': 6, '# Samples': 3270},
# 'test': {'DEBW013': 1066, 'DEBW076': 696, 'DEBW087': 1080, 'DEBW107': 1080, 'DEBY081': 700,
# '# Stations': 6, '# Samples': 4622}}
# df = pd.DataFrame.from_dict(data_dict)
# return df
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment