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.
42 lines
1.1 KiB
42 lines
1.1 KiB
15 years ago
|
#!/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())
|