def test_waterfall(test_data, keys, kwargs):
data_files = (test_data[key] for key in keys)
cli.waterfall(*data_files, **kwargs, show_fig=False)
plt.close()
def test_waterfall_exception():
with pytest.raises(ValueError):
cli.waterfall(*tuple())
plt.close()
def test_waterfall(db, keys, kwargs):
data_files = (db[key] for key in keys)
cli.waterfall(*data_files, **kwargs)
plt.close()