def test():
# 3 cards on flat table
cards_3 = cv2.imread('images/set-3-texture.jpg')
# 5 cards at an angle
cards_5 = cv2.imread('images/set-5-random.jpg')
thresh_3 = s.get_binary(cards_3)
contours = s.find_contours(thresh_3, 3)
assert len(s.transform_cards(cards_3, contours, 3)) == 3
res5 = s.detect_cards(cards_5)
assert res5 is not None and len(res5) == 5
res3 = s.detect_cards(cards_3)
assert res3 is not None and len(res3) == 3
for i in range(len(res5)):
c = res5[i]
# util.show(c, 'card')
cv2.imwrite('images/cards/card-5-%d.jpg' % i, c)
for i in range(len(res3)):
c = res3[i]
# util.show(c, 'card')
cv2.imwrite('images/cards/card-3-%d.jpg' % i, c)
# for cards detected, get properties
for link in os.listdir('images/cards'):
img = cv2.imread('images/cards/%s' % link)
test_props(img)
print 'tests pass'
def main():
# 3 of the 12 set that's bad
cards_3_bad = cv2.imread('images/set-3-bad.jpg')
thresh_3bad = s.get_binary(cards_3_bad)
res3bad = s.detect_cards(cards_3_bad)
assert res3bad is not None and len(res3bad) == 3
# 12 cards
cards_12 = cv2.imread('images/set-12-random-2sets-sm.jpg')
thresh_12bad = s.get_binary(cards_12)
res12bad = s.detect_cards(cards_12, draw_rects=False)
util.show(cards_12)
# Subset of 3, with the 1 problem card
cards = res12bad
for i in range(len(cards)):
card = cards[i]
# test_props(card)
cv2.imwrite('images/cards/card-12-%02d.jpg' % i, card)
props = s.get_cards_properties(res12bad)
s.pretty_print_properties(props)
g = game(cards=props)
sets = g.play(True)
if sets:
print '\nFound sets:'
for st in sets:
just_props = [e for i, e in st]
print just_props
s.pretty_print_properties(just_props)
print('---')
else:
print 'no sets :('
print 'tests pass'
