test: add simple dilation tests

seung-lab · Dec 15, 2023 · 2182a0c · 2182a0c
1 parent f4b8188
commit 2182a0c
Showing 1 changed file with 23 additions and 0 deletions.
diff --git a/automated_test.py b/automated_test.py
@@ -76,3 +76,26 @@ def test_spherical_close():
 	res = fastmorph.spherical_close(labels, radius=1)
 	assert res[5,5,5] == True
 
+
+def test_dilate():
+	labels = np.zeros((3,3,3), dtype=bool)
+
+	out = fastmorph.dilate(labels)
+	assert not np.any(out)
+
+	labels[1,1,1] = True
+
+	out = fastmorph.dilate(labels)
+	assert np.all(out)
+
+	labels = np.zeros((3,3,3), dtype=bool)
+	labels[0,0,0] = True
+	out = fastmorph.dilate(labels)
+
+	ans = np.zeros((3,3,3), dtype=bool)
+	ans[:2,:2,:2] = True
+
+	assert np.all(out == ans)
+
+
+