model/openface_model_test.py

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