class HorizonTest(unittest.TestCase):
#setUp
def setUp(self):
#create Grayscale picture for test
testColorImg1 = cv.LoadImage("./HorizonTestData/test1.jpg",cv.CV_LOAD_IMAGE_COLOR)
cv.Smooth(testColorImg1,testColorImg1,cv.CV_GAUSSIAN,13,0,0,0)
testGrayImage1 = cv.CreateImage((testColorImg1.width,testColorImg1.height),cv.IPL_DEPTH_8U,1)
cv.CvtColor(testColorImg1,testGrayImage1,cv.CV_BGR2GRAY)
cv.EqualizeHist(testGrayImage1,testGrayImage1)
#create instance
self.horizon = Horizon(testGrayImage1)
self.horizon.test(testColorImg1)
# def testAll(self):
#インスタンスが生成されていることを確認する
#Tests exsit of instance
# self.assertIsInstance(self.horizon,Horizon)
#Tests calcHorizon
def testDetectHough(self):
self.assertIsInstance(self.horizon,Horizon)
self.horizon.detectHorizon()
def getHorizon(self): try: self.getCameraCVPict() self.convertGray() horizon = Horizon(self.__grayCameraPict) horizon.detectHorizon() degree = horizon.getHorizon() print "degree:"+str(degree) return degree except: sys.exit(sys.exc_info()[:2])
