Exemplo n.º 1
0
    def test_add_blacklist_item(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_REC_DATA_DIR_PATH_KEY: face_rec_data}

        item = 'Al_telefono'

        fm = FaceModels(params)

        fm.add_blacklist_item(item)

        blacklist = fm.get_blacklist()

        self.assertIn(item, blacklist)