mirror of
https://github.com/wassname/scikit-image.git
synced 2026-07-04 01:58:43 +08:00
Stefan van der Walt
31 lines
606 B
Python
31 lines
606 B
Python
import numpy as np
|
|
from numpy.testing import *
|
|
|
|
from scikits.image.transform import *
|
|
|
|
def test_hough():
|
|
# Generate a test image
|
|
img = np.zeros((100, 150), dtype=bool)
|
|
img[30, :] = 1
|
|
img[:, 65] = 1
|
|
img[35:45, 35:50] = 1
|
|
for i in range(90):
|
|
img[i, i] = 1
|
|
|
|
out, angles, d = hough(img)
|
|
|
|
assert_equal(out.max(), 100)
|
|
assert_equal(len(angles), 180)
|
|
|
|
def test_hough_angles():
|
|
img = np.zeros((10, 10))
|
|
img[0, 0] = 1
|
|
|
|
out, angles, d = hough(img, np.linspace(0, 360, 10))
|
|
|
|
assert_equal(len(angles), 10)
|
|
|
|
if __name__ == "__main__":
|
|
run_module_suite()
|
|
|