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()