assert equal scalar, seqeunce output of hough circle

This commit is contained in:
Julius Bier Kirekgaard
2015-08-30 15:13:45 +01:00
parent 9a4a3c674e
commit 8c5ce8504a
@@ -140,9 +140,11 @@ def test_hough_circle():
y, x = circle_perimeter(y_0, x_0, radius)
img[x, y] = 1
out = tf.hough_circle(img, radius)
out = tf.hough_circle(img, [radius])
out1 = tf.hough_circle(img, radius)
out2 = tf.hough_circle(img, [radius])
assert_equal(out1,out2)
out = tf.hough_circle(img, np.array([radius], dtype=np.intp))
assert_equal(out,out1)
x, y = np.where(out[0] == out[0].max())
assert_equal(x[0], x_0)
assert_equal(y[0], y_0)