def test_decode(self):
a= Square.SQ_SIZE
b = a*(Square.GRID_SIZE+3)
rect = [(a,a),(b,a),(b,b),(a,b)]
m_d = SquareDetector(Square.generate(1),0)
for i in range(32):
img = Square.generate(i)
sq = Square(i,m_d)
m_d.gray_img=img
a,b = sq._decode_rect(rect)
self.assertTrue((sq.TOP_LEFT,i)==(a,b),"Wrong %d: (%d,%d)"%(i,a,b))
cv.Transpose(img, img)
cv.Flip(img)
a,b = sq._decode_rect(rect)
self.assertTrue((sq.BOT_LEFT,i)==(a,b),"Wrong %d: (%d,%d)"%(i,a,b))
cv.Transpose(img, img)
cv.Flip(img)
a,b = sq._decode_rect(rect)
self.assertTrue((sq.BOT_RIGHT,i)==(a,b),"Wrong %d: (%d,%d)"%(i,a,b))
cv.Transpose(img, img)
cv.Flip(img)
a,b = sq._decode_rect(rect)
self.assertTrue((sq.TOP_RIGHT,i)==(a,b),"Wrong %d: (%d,%d)"%(i,a,b))
m_d.flip_H=True
for i in range(32):
img = Square.generate(i)
sq = Square(i,m_d)
m_d.gray_img=img
cv.Flip(img,img,1)
_,b = sq._decode_rect(rect)
self.assertTrue(i==b,"Wrong Flip %d: (%d)"%(i,b))
