def test_save_graphviz_different_filename_output(file_name, format,
interactive,
fitted_tree_estimators,
tmpdir,
has_minimal_dependencies):
if not has_minimal_dependencies:
est_class, _ = fitted_tree_estimators
src = visualize_decision_tree(estimator=est_class,
filled=True,
max_depth=3)
filepath = os.path.join(str(tmpdir), f'{file_name}')
no_output_ = save_plot(fig=src,
filepath=filepath,
format=format,
interactive=interactive,
return_filepath=False)
output_ = save_plot(fig=src,
filepath=filepath,
format=format,
interactive=interactive,
return_filepath=True)
assert not no_output_
assert os.path.exists(output_)
assert isinstance(output_, str)
assert os.path.basename(output_) == 'example_plot.png'
def test_save_graphviz_invalid_filepath(file_name, format, interactive, fitted_tree_estimators, tmpdir, has_minimal_dependencies):
if not has_minimal_dependencies:
est_class, _ = fitted_tree_estimators
src = visualize_decision_tree(estimator=est_class, filled=True, max_depth=3)
filepath = f'{file_name}.{format}'
with pytest.raises(ValueError, match="Specified filepath is not writeable"):
save_plot(fig=src, filepath=filepath, format=format, interactive=interactive, return_filepath=False)