class TestSubject(unittest.TestCase):
class TestData:
def __init__(self, name, user_data):
self.name = name
self.user_data = user_data
self.verbose = True
def setUp(self):
self.recognition = Recognition()
self.face_log = FaceLog()
# Test on some known videos
def create_subjects(self):
print "Creating some subjects"
test_data = [
TestSubject.TestData(name='Kieron', user_data='user data'),
TestSubject.TestData(name='Katie', user_data='a lady')
]
for this_test in test_data:
print "** Testing {0} **".format(this_test.name)
subject = self.recognition.create_subject(name=this_test.name, user_data=this_test.user_data)
print 'created subject {name} {subject_id}'.format(name=subject['name'], subject_id=subject['subject_id'])
self.assertIsNotNone(subject['subject_id'])
def list_subjects(self):
print "List all the subjects"
subjects = self.recognition.list_subjects()
self.assertIsNotNone(subjects)
for subject in subjects:
print '{0} has id {1} and thumbnail {2}'.format(subject['name'], subject['subject_id'], subject['thumbnail'])
self.recognition.download_with_authentication(subject['thumbnail'], '{}.jpg'.format(subject['subject_id']))
def delete_subjects(self):
subjects = self.recognition.list_subjects()
for subject in subjects:
print 'Going to delete {0}'.format(subject['subject_id'])
deleted_id = self.recognition.delete_subject(subject['subject_id'])
self.assertEqual(subject['subject_id'], deleted_id)
