def test_clear_input_no_confirm(): f = ImageFeaturizer() with pytest.raises(ValueError): f.clear_input()
def test_clear_input(): f = ImageFeaturizer() f.featurize(save_features=True, omit_time=True, omit_model=True, omit_depth=True, omit_output=True, **LOAD_DATA_ARGS) f.clear_input(confirm=True) compare_empty_input(f)