def test_remove_face(self):
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.delete_models()
self.test_add_face()
training_set_path = os.path.join(
fm._data_dir_path, c.TRAINING_SET_DIR)
aligned_faces_path = os.path.join(
training_set_path, c.ALIGNED_FACES_DIR)
label = 3812
subject_path = os.path.join(aligned_faces_path, str(label))
im_name = ''
for im in os.listdir(subject_path):
im_name = im
ok = fm.remove_face(label, im_name)
self.assertTrue(ok)
def test_remove_face_not_removed(self):
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.delete_models()
self.test_add_face()
label = 1812
im_name = 'Dummy image'
ok = fm.remove_face(label, im_name)
self.assertFalse(ok)
