def test_create_timestep_7days(): dfs0File = r"zeros_ones.dfs0" data = [] nt = 10 d1 = np.zeros(nt) data.append(d1) d2 = np.ones(nt) data.append(d2) names = ["Zeros", "Ones"] dfs = dfs0.dfs0() dfs.create(filename=dfs0File, data=data, names=names, title="Zeros and ones", timeseries_unit=TimeStep.DAY, dt=7) assert True res = dfs.read(dfs0File) dt = res.time[1] - res.time[0] assert dt == timedelta(days=7) os.remove(dfs0File)
def test_create_equidistant_calendar(): dfs0file = r'random.dfs0' d1 = np.random.random([1000]) d2 = np.random.random([1000]) data = [] data.append(d1) data.append(d2) start_time = datetime.datetime(2017, 1, 1) timeseries_unit = 1402 title = 'Hello Test' names = ['VarFun01', 'NotFun'] variable_type = [100000, 100000] unit = [1000, 1000] data_value_type = [0, 1] dt = 5 dfs = dfs0.dfs0() dfs.create(filename=dfs0file, data=data, start_time=start_time, timeseries_unit=timeseries_unit, dt=dt, names=names, title=title, variable_type=variable_type, unit=unit, data_value_type=data_value_type) os.remove(dfs0file) assert True
def test_read_dfs0_to_matrix(): dfs0file = r'tests/testdata/random.dfs0' dfs = dfs0.dfs0() (data, t, names) = dfs.read(filename=dfs0file) assert len(data) == 2
def test_create_non_equidistant_calendar(): dfs0file = r'neq.dfs0' d1 = np.random.random([1000]) d2 = np.random.random([1000]) data = [] data.append(d1) data.append(d2) start_time = datetime.datetime(2017, 1, 1) time_vector = [] for i in range(1000): time_vector.append(start_time + datetime.timedelta(hours=i*0.1)) title = 'Hello Test' names = ['VarFun01', 'NotFun'] variable_type = [100000, 100000] unit = [1000, 1000] data_value_type = [0, 1] dfs = dfs0.dfs0() dfs.create(filename=dfs0file, data=data, datetimes=time_vector, names=names, title=title, variable_type=variable_type, unit=unit, data_value_type=data_value_type) assert True os.remove(dfs0file)
def test_read_dfs0_to_pandas(): dfs0file = r'tests/testdata/random.dfs0' dfs = dfs0.dfs0() pd = dfs.read_to_pandas(dfs0file) assert np.isnan(pd[pd.columns[0]][2])
def test_read_dfs0_single_item(): dfs0file = r'tests/testdata/random.dfs0' dfs = dfs0.dfs0() (data, t, names) = dfs.read(dfs0file, item_numbers=[1]) assert len(data) == 1
def test_read_dfs0_to_pandas_single_item(): dfs0file = r'tests/testdata/random.dfs0' dfs = dfs0.dfs0() df = dfs.read_to_pandas(dfs0file, item_numbers=[1]) assert df.shape[1] == 1
def test_read_dfs0_single_item_named_access(): dfs0file = r'tests/testdata/random.dfs0' dfs = dfs0.dfs0() res = dfs.read(dfs0file, item_numbers=[1]) data = res.data assert len(data) == 1
def test_simple_create(): dfs0File = r"simple.dfs0" data = [] nt = 100 d = np.random.random([nt]) data.append(d) dfs = dfs0.dfs0() dfs.create(filename=dfs0File, data=data) assert True os.remove(dfs0File)
def test_multiple_create(): dfs0File = r"zeros_ones.dfs0" data = [] nt = 10 d1 = np.zeros(nt) data.append(d1) d2 = np.ones(nt) data.append(d2) names = ["Zeros", "Ones"] dfs = dfs0.dfs0() dfs.create(filename=dfs0File, data=data, names=names, title="Zeros and ones") assert True os.remove(dfs0File)