Fix canny test to reflect new error type.

This commit is contained in:
Steven Silvester
2014-09-20 20:09:44 -05:00
parent 04f91021a9
commit bcac1b1a6d
+1 -1
View File
@@ -60,7 +60,7 @@ class TestCanny(unittest.TestCase):
self.assertTrue(point_count < 1600)
def test_image_shape(self):
self.assertRaises(TypeError, F.canny, np.zeros((20, 20, 20)), 4, 0, 0)
self.assertRaises(ValueError, F.canny, np.zeros((20, 20, 20)), 4, 0, 0)
def test_mask_none(self):
result1 = F.canny(np.zeros((20, 20)), 4, 0, 0, np.ones((20, 20), bool))