def test_require_variable_top_level(test_df): exp = pd.DataFrame([["model_a", "scen_b"]], columns=META_IDX) # checking that the return-type is correct obs = require_variable(test_df, variable="Primary Energy|Coal") pdt.assert_frame_equal(obs, exp) assert list(test_df["exclude"]) == [False, False] # checking exclude on fail obs = require_variable(test_df, variable="Primary Energy|Coal", exclude_on_fail=True) pdt.assert_frame_equal(obs, exp) assert list(test_df["exclude"]) == [False, True]
def test_require_variable_top_level(test_df): obs = require_variable(test_df, variable='Primary Energy|Coal', exclude_on_fail=True) assert len(obs) == 1 assert obs.loc[0, 'scenario'] == 'scen_b' assert list(test_df['exclude']) == [False, True]