mirror of https://github.com/opencv/opencv.git
Open Source Computer Vision Library
https://opencv.org/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
41 lines
1.1 KiB
41 lines
1.1 KiB
#!/usr/bin/env python |
|
import cvtestutils |
|
import unittest |
|
from cv import * |
|
|
|
class moments_test(unittest.TestCase): |
|
def setUp(self): |
|
# create an image |
|
img = cvCreateMat(100,100,CV_8U); |
|
|
|
cvZero( img ) |
|
# draw a rectangle in the middle |
|
cvRectangle( img, cvPoint( 25, 25 ), cvPoint( 75, 75 ), CV_RGB(255,255,255), -1 ); |
|
|
|
self.img = img |
|
|
|
# create the storage area |
|
self.storage = cvCreateMemStorage (0) |
|
|
|
# find the contours |
|
nb_contours, self.contours = cvFindContours (img, |
|
self.storage, |
|
sizeof_CvContour, |
|
CV_RETR_LIST, |
|
CV_CHAIN_APPROX_SIMPLE, |
|
cvPoint (0,0)) |
|
|
|
def test_cvMoments_CvMat( self ): |
|
m = CvMoments() |
|
cvMoments( self.img, m, 1 ) |
|
def test_cvMoments_CvSeq( self ): |
|
m = CvMoments() |
|
# Now test with CvSeq |
|
for contour in self.contours.hrange(): |
|
cvMoments( contour, m, 1 ) |
|
|
|
def suite(): |
|
return unittest.TestLoader().loadTestsFromTestCase(moments_test) |
|
|
|
if __name__ == '__main__': |
|
unittest.TextTestRunner(verbosity=2).run(suite())
|
|
|