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()