Exemplo n.º 1
0
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]
Exemplo n.º 2
0
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]
Exemplo n.º 3
0
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]
Exemplo n.º 4
0
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]
Exemplo n.º 5
0
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]