def test_separate_watershed(vdf_image, min_distance, min_size, max_size,
max_number_of_grains, exclude_border, threshold,
sep_shape_expt):
sep = separate_watershed(vdf_image, min_distance=min_distance,
min_size=min_size, max_size=max_size,
max_number_of_grains=max_number_of_grains,
exclude_border=exclude_border, threshold=threshold)
np.testing.assert_equal(sep.shape, sep_shape_expt)
def test_separate_watershed(
vdf_image,
min_distance,
min_size,
max_size,
max_number_of_grains,
exclude_border,
threshold,
sep_shape_expt,
):
sep = separate_watershed(
vdf_image,
min_distance=min_distance,
min_size=min_size,
max_size=max_size,
max_number_of_grains=max_number_of_grains,
exclude_border=exclude_border,
threshold=threshold,
)
# we don't care how many clusters we generate
assert sep.shape[1:] == sep_shape_expt[1:]