mirror of
https://github.com/wassname/scikit-image.git
synced 2026-07-27 11:27:08 +08:00
added unit-tests for existing functions. and prelimenary docstrings
This commit is contained in:
sccolbert
Binary file not shown.
|
After Width: | Height: | Size: 32 KiB |
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
|
After Width: | Height: | Size: 11 KiB |
@@ -0,0 +1,152 @@
|
||||
# test for the opencv_cv extension module
|
||||
|
||||
import numpy as np
|
||||
from numpy.testing import *
|
||||
|
||||
try:
|
||||
from scikits.image.opencv import *
|
||||
OPENCV_LIBS_NOTFOUND = False
|
||||
except:
|
||||
OPENCV_LIBS_NOTFOUND = True
|
||||
|
||||
@dec.skipif(OPENCV_LIBS_NOTFOUND, 'Skipping OpenCV test because OpenCV libs were not found')
|
||||
def test_cvSobel():
|
||||
lena_RGB_U8 = np.load('data/lena_RGB_U8.npy')
|
||||
lena_GRAY_U8 = np.load('data/lena_GRAY_U8.npy')
|
||||
lena_SOBEL_INT16 = np.load('data/lena_SOBEL_INT16.npy')
|
||||
sobel = cvSobel(lena_GRAY_U8)
|
||||
assert_array_equal(sobel, lena_SOBEL_INT16)
|
||||
assert_raises(Exception, cvSobel, lena_RGB_U8)
|
||||
assert_raises(Exception, cvSobel, (lena_GRAY_U8, lena_GRAY_U8))
|
||||
assert_raises(Exception, cvSobel, lena_GRAY_U8, {'aperture_size':1})
|
||||
assert_raises(Exception, cvSobel, lena_GRAY_U8, {'aperture_size':2})
|
||||
assert_raises(Exception, cvSobel, lena_GRAY_U8, {'aperture_size':4})
|
||||
assert_raises(Exception, cvSobel, lena_GRAY_U8, {'aperture_size':6})
|
||||
assert_raises(Exception, cvSobel, lena_GRAY_U8, {'aperture_size':8})
|
||||
# any need to keep going? only valid apertures are 3, 5, and 7
|
||||
# and testing the validity would require saving new images for each one
|
||||
|
||||
@dec.skipif(OPENCV_LIBS_NOTFOUND, 'Skipping OpenCV test because OpenCV libs were not found')
|
||||
def test_cvLaplace():
|
||||
lena_RGB_U8 = np.load('data/lena_RGB_U8.npy')
|
||||
lena_GRAY_U8 = np.load('data/lena_GRAY_U8.npy')
|
||||
lena_LAPLACE_INT16 = np.load('data/lena_LAPLACE_INT16.npy')
|
||||
laplace = cvLaplace(lena_GRAY_U8)
|
||||
assert_array_equal(laplace, lena_LAPLACE_INT16)
|
||||
assert_raises(Exception, cvLaplace, lena_RGB_U8)
|
||||
assert_raises(Exception, cvLaplace, (lena_GRAY_U8, lena_GRAY_U8))
|
||||
assert_raises(Exception, cvLaplace, lena_GRAY_U8, {'aperture_size':1})
|
||||
assert_raises(Exception, cvLaplace, lena_GRAY_U8, {'aperture_size':2})
|
||||
assert_raises(Exception, cvLaplace, lena_GRAY_U8, {'aperture_size':4})
|
||||
assert_raises(Exception, cvLaplace, lena_GRAY_U8, {'aperture_size':6})
|
||||
assert_raises(Exception, cvLaplace, lena_GRAY_U8, {'aperture_size':8})
|
||||
# any need to keep going? only valid apertures are 3, 5, and 7
|
||||
# and testing the validity would require saving new images for each one
|
||||
|
||||
@dec.skipif(OPENCV_LIBS_NOTFOUND, 'Skipping OpenCV test because OpenCV libs were not found')
|
||||
def test_cvCanny():
|
||||
lena_RGB_U8 = np.load('data/lena_RGB_U8.npy')
|
||||
lena_GRAY_U8 = np.load('data/lena_GRAY_U8.npy')
|
||||
lena_CANNY_U8 = np.load('data/lena_CANNY_U8.npy')
|
||||
canny = cvCanny(lena_GRAY_U8)
|
||||
assert_array_equal(canny, lena_CANNY_U8)
|
||||
assert_raises(Exception, cvCanny, lena_RGB_U8)
|
||||
assert_raises(Exception, cvCanny, (lena_GRAY_U8, lena_GRAY_U8))
|
||||
assert_raises(Exception, cvCanny, lena_GRAY_U8, {'aperture_size':1})
|
||||
assert_raises(Exception, cvCanny, lena_GRAY_U8, {'aperture_size':2})
|
||||
assert_raises(Exception, cvCanny, lena_GRAY_U8, {'aperture_size':4})
|
||||
assert_raises(Exception, cvCanny, lena_GRAY_U8, {'aperture_size':6})
|
||||
assert_raises(Exception, cvCanny, lena_GRAY_U8, {'aperture_size':8})
|
||||
# any need to keep going? only valid apertures are 3, 5, and 7
|
||||
# and testing the validity would require saving new images for each one
|
||||
|
||||
@dec.skipif(OPENCV_LIBS_NOTFOUND, 'Skipping OpenCV test because OpenCV libs were not found')
|
||||
def test_cvPreCornerDetect():
|
||||
lena_RGB_U8 = np.load('data/lena_RGB_U8.npy')
|
||||
lena_GRAY_U8 = np.load('data/lena_GRAY_U8.npy')
|
||||
lena_GRAY_FLOAT32 = np.load('data/lena_GRAY_FLOAT32.npy')
|
||||
lena_PRECORNERDETECT_U8 = np.load('data/lena_PRECORNERDETECT_U8.npy')
|
||||
lena_PRECORNERDETECT_FLOAT32 = np.load('data/lena_PRECORNERDETECT_FLOAT32.npy')
|
||||
pcd8 = cvPreCornerDetect(lena_GRAY_U8)
|
||||
pcd32 = cvPreCornerDetect(lena_GRAY_FLOAT32)
|
||||
assert_array_almost_equal(pcd8, lena_PRECORNERDETECT_U8)
|
||||
assert_array_almost_equal(pcd32, lena_PRECORNERDETECT_FLOAT32)
|
||||
assert_raises(Exception, cvPreCornerDetect, lena_RGB_U8)
|
||||
assert_raises(Exception, cvPreCornerDetect, (lena_GRAY_U8, lena_GRAY_U8))
|
||||
assert_raises(Exception, cvPreCornerDetect, (lena_GRAY_U8, lena_PRECORNERDETECT_U8))
|
||||
assert_raises(Exception, cvPreCornerDetect, lena_GRAY_U8, {'aperture_size':1})
|
||||
assert_raises(Exception, cvPreCornerDetect, lena_GRAY_U8, {'aperture_size':2})
|
||||
assert_raises(Exception, cvPreCornerDetect, lena_GRAY_U8, {'aperture_size':4})
|
||||
assert_raises(Exception, cvPreCornerDetect, lena_GRAY_U8, {'aperture_size':6})
|
||||
assert_raises(Exception, cvPreCornerDetect, lena_GRAY_U8, {'aperture_size':8})
|
||||
|
||||
@dec.skipif(OPENCV_LIBS_NOTFOUND, 'Skipping OpenCV test because OpenCV libs were not found')
|
||||
def test_cvCornerEigenValsAndVecs():
|
||||
lena_RGB_U8 = np.load('data/lena_RGB_U8.npy')
|
||||
lena_GRAY_U8 = np.load('data/lena_GRAY_U8.npy')
|
||||
lena_GRAY_FLOAT32 = np.load('data/lena_GRAY_FLOAT32.npy')
|
||||
lena_CEVAV_U8 = np.load('data/lena_CEVAV_U8.npy')
|
||||
lena_CEVAV_FLOAT32 = np.load('data/lena_CEVAV_FLOAT32.npy')
|
||||
cevav8 = cvCornerEigenValsAndVecs(lena_GRAY_U8)
|
||||
cevav32 = cvCornerEigenValsAndVecs(lena_GRAY_FLOAT32)
|
||||
assert_array_almost_equal(cevav8, lena_CEVAV_U8)
|
||||
assert_array_almost_equal(cevav32, lena_CEVAV_FLOAT32)
|
||||
assert_raises(Exception, cvCornerEigenValsAndVecs, lena_RGB_U8)
|
||||
assert_raises(Exception, cvCornerEigenValsAndVecs, lena_GRAY_U8, {'aperture_size':1})
|
||||
assert_raises(Exception, cvCornerEigenValsAndVecs, lena_GRAY_U8, {'aperture_size':2})
|
||||
assert_raises(Exception, cvCornerEigenValsAndVecs, lena_GRAY_U8, {'aperture_size':4})
|
||||
assert_raises(Exception, cvCornerEigenValsAndVecs, lena_GRAY_U8, {'aperture_size':6})
|
||||
assert_raises(Exception, cvCornerEigenValsAndVecs, lena_GRAY_U8, {'aperture_size':8})
|
||||
|
||||
@dec.skipif(OPENCV_LIBS_NOTFOUND, 'Skipping OpenCV test because OpenCV libs were not found')
|
||||
def test_cvCornerMinEigenVal():
|
||||
lena_RGB_U8 = np.load('data/lena_RGB_U8.npy')
|
||||
lena_GRAY_U8 = np.load('data/lena_GRAY_U8.npy')
|
||||
lena_GRAY_FLOAT32 = np.load('data/lena_GRAY_FLOAT32.npy')
|
||||
lena_CMEV_U8 = np.load('data/lena_CMEV_U8.npy')
|
||||
lena_CMEV_FLOAT32 = np.load('data/lena_CMEV_FLOAT32.npy')
|
||||
cmev8 = cvCornerMinEigenVal(lena_GRAY_U8)
|
||||
cmev32 = cvCornerMinEigenVal(lena_GRAY_FLOAT32)
|
||||
assert_array_almost_equal(cmev8, lena_CMEV_U8)
|
||||
assert_array_almost_equal(cmev32, lena_CMEV_FLOAT32)
|
||||
assert_raises(Exception, cvCornerMinEigenVal, lena_RGB_U8)
|
||||
assert_raises(Exception, cvCornerMinEigenVal, lena_GRAY_U8, {'aperture_size':1})
|
||||
assert_raises(Exception, cvCornerMinEigenVal, lena_GRAY_U8, {'aperture_size':2})
|
||||
assert_raises(Exception, cvCornerMinEigenVal, lena_GRAY_U8, {'aperture_size':4})
|
||||
assert_raises(Exception, cvCornerMinEigenVal, lena_GRAY_U8, {'aperture_size':6})
|
||||
assert_raises(Exception, cvCornerMinEigenVal, lena_GRAY_U8, {'aperture_size':8})
|
||||
|
||||
@dec.skipif(OPENCV_LIBS_NOTFOUND, 'Skipping OpenCV test because OpenCV libs were not found')
|
||||
def test_cvCornerHarris():
|
||||
lena_RGB_U8 = np.load('data/lena_RGB_U8.npy')
|
||||
lena_GRAY_U8 = np.load('data/lena_GRAY_U8.npy')
|
||||
lena_GRAY_FLOAT32 = np.load('data/lena_GRAY_FLOAT32.npy')
|
||||
lena_HARRIS_U8 = np.load('data/lena_HARRIS_U8.npy')
|
||||
lena_HARRIS_FLOAT32 = np.load('data/lena_HARRIS_FLOAT32.npy')
|
||||
hc8 = cvCornerHarris(lena_GRAY_U8)
|
||||
hc32 = cvCornerHarris(lena_GRAY_FLOAT32)
|
||||
assert_array_almost_equal(hc8, lena_HARRIS_U8)
|
||||
assert_array_almost_equal(hc32, lena_HARRIS_FLOAT32)
|
||||
assert_raises(Exception, cvCornerHarris, lena_RGB_U8)
|
||||
assert_raises(Exception, cvCornerHarris, lena_GRAY_U8, {'aperture_size':1})
|
||||
assert_raises(Exception, cvCornerHarris, lena_GRAY_U8, {'aperture_size':2})
|
||||
assert_raises(Exception, cvCornerHarris, lena_GRAY_U8, {'aperture_size':4})
|
||||
assert_raises(Exception, cvCornerHarris, lena_GRAY_U8, {'aperture_size':6})
|
||||
assert_raises(Exception, cvCornerHarris, lena_GRAY_U8, {'aperture_size':8})
|
||||
|
||||
@dec.skipif(OPENCV_LIBS_NOTFOUND, 'Skipping OpenCV test because OpenCV libs were not found')
|
||||
def test_cvSmooth():
|
||||
lena_RGB_U8 = np.load('data/lena_RGB_U8.npy')
|
||||
lena_GRAY_U8 = np.load('data/lena_GRAY_U8.npy')
|
||||
lena_GAUSSRGB_U8 = np.load('data/lena_GAUSSRGB_U8.npy')
|
||||
lena_GAUSSGRAY_U8 = np.load('data/lena_GAUSSGRAY_U8.npy')
|
||||
gaussrgb = cvSmooth(lena_RGB_U8)
|
||||
gaussgray = cvSmooth(lena_GRAY_U8)
|
||||
assert_array_equal(gaussrgb, lena_GAUSSRGB_U8)
|
||||
assert_array_equal(gaussgray, lena_GAUSSGRAY_U8)
|
||||
assert_raises(Exception, cvSmooth, (lena_RGB_U8), {'smoothtype': CV_BLUR_NO_SCALE})
|
||||
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
run_module_suite()
|
||||
Reference in New Issue
Block a user