def test_shape_mismatch():
image = np.ones((3, 3))
label = np.ones((2, 2))
testing.assert_raises(ValueError, label2rgb, image, label)
微信
公众号
扫码关注公众号