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')