def test_get_tag(self):
self.test_add_face()
base_path = os.path.join('..', 'test_files', 'face_models')
face_rec_data = os.path.abspath(os.path.join(base_path, 'face_rec_data'))
params = {c.GLOBAL_FACE_MODELS_MIN_DIFF_KEY: -1,
c.GLOBAL_FACE_REC_DATA_DIR_PATH_KEY: face_rec_data}
fm = FaceModels(params)
fm.load_models()
tag = fm.get_tag(3812)
self.assertEqual(tag, 'Mameli_Giacomo')
def test_change_tag_to_label(self):
self.test_add_face()
base_path = os.path.join('..', 'test_files', 'face_models')
face_rec_data = os.path.abspath(os.path.join(base_path, 'face_rec_data'))
params = {c.GLOBAL_FACE_REC_DATA_DIR_PATH_KEY: face_rec_data}
fm = FaceModels(params)
label = 3812
new_tag = 'Giacomo_Mameli'
fm.change_tag_to_label(label, new_tag)
tag = fm.get_tag(label)
self.assertEqual(tag, 'Giacomo_Mameli')
