Test several more branch points in draw

jni · Jan 9, 2018 · 73456fd · 73456fd
1 parent d75f4d5
commit 73456fd
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/skan/test/test_draw.py b/skan/test/test_draw.py
@@ -41,8 +41,8 @@ def test_stats(test_skeleton):
 
 def test_overlay_skeleton(test_image, test_skeleton):
     draw.overlay_skeleton_2d(test_image, test_skeleton)
-    draw.overlay_skeleton_2d(test_image, test_skeleton,
-                                     image_cmap='viridis')
+    draw.overlay_skeleton_2d(test_image, test_skeleton, image_cmap='viridis')
+    draw.overlay_skeleton_2d(test_image, test_skeleton, dilate=1)
 
 
 def test_overlay_euclidean_skeleton(test_image, test_stats):
@@ -75,3 +75,6 @@ def test_networkx_plot():
                                    axis=axes[0])
     draw.overlay_skeleton_networkx(g1, c1, image=_testdata.skeleton1,
                                    axis=axes[1])
+    # test axis=None and additional coordinates
+    c2 = np.concatenate((c1, np.random.random(c1[:1].shape)), axis=0)
+    draw.overlay_skeleton_networkx(g1, c2, image=_testdata.skeleton1)