def test_validate_top_level(test_df): obs = validate(test_df, criteria={'Primary Energy': { 'up': 6 }}, exclude_on_fail=True, variable='Primary Energy') pdt.assert_frame_equal(obs, test_df.data[5:6].reset_index(drop=True)) assert list(test_df['exclude']) == [False, True]
def test_validate_top_level(meta_df): obs = validate(meta_df, criteria={'Primary Energy': { 'up': 6.0 }}, exclude_on_fail=True, variable='Primary Energy') assert len(obs) == 1 assert obs['year'].values[0] == 2010 assert list(meta_df['exclude']) == [False, True]
def test_validate_top_level(test_df): obs = validate(test_df, criteria={'Primary Energy': {'up': 6.0}}, exclude_on_fail=True, variable='Primary Energy') assert len(obs) == 1 if 'year' in test_df.data: assert obs['year'].values[0] == 2010 else: exp_time = pd.to_datetime(datetime.datetime(2010, 7, 21)) assert (pd.to_datetime(obs['time'].values[0]).date() == exp_time) assert list(test_df['exclude']) == [False, True]
def test_validate_top_level(test_df): obs = validate(test_df, criteria={'Primary Energy': {'up': 6.0}}, exclude_on_fail=True, variable='Primary Energy') assert len(obs) == 1 if 'year' in test_df._data.index.names: assert obs.index.get_level_values('year').values[0] == 2010 else: exp_time = datetime.datetime(2010, 7, 21).date() obs_time = obs.index.get_level_values('time')[0].date() assert exp_time == obs_time assert list(test_df['exclude']) == [False, True]
def test_validate_top_level(test_df): obs = validate( test_df, criteria={"Primary Energy": { "up": 6 }}, exclude_on_fail=True, variable="Primary Energy", ) pdt.assert_frame_equal(obs, test_df.data[5:6].reset_index(drop=True)) assert list(test_df["exclude"]) == [False, True]