import pytest from openface_model import OpenFaceModel @pytest.fixture() def model(): return OpenFaceModel() @pytest.mark.skip(reason="Not implemented") def test_embed(model): image = open("../testdata/face_pic.jpeg", "rb").read() aligned_face = model.preprocess(image) rep = model.embed(aligned_face) assert rep.shape == (128,)