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]
