Class _Harness

backend/python/insightface/test.py:120–160

class _Harness:
    def __init__(self, servicer: BackendServicer) -> None:
        self.svc = servicer

    def health(self):
        return self.svc.Health(backend_pb2.HealthMessage(), _FakeContext())

    def load(self, options: list[str], model_path: str = ""):
        return self.svc.LoadModel(
            backend_pb2.ModelOptions(Model="test", Options=options, ModelPath=model_path),
            _FakeContext(),
        )

    def detect(self, img_b64: str):
        return self.svc.Detect(backend_pb2.DetectOptions(src=img_b64), _FakeContext())

    def embed(self, img_b64: str):
        ctx = _FakeContext()
        res = self.svc.Embedding(
            backend_pb2.PredictOptions(Images=[img_b64]),
            ctx,
        )
        return res, ctx

    def verify(self, a: str, b: str, threshold: float = 0.0, anti_spoofing: bool = False):
        ctx = _FakeContext()
        res = self.svc.FaceVerify(
            backend_pb2.FaceVerifyRequest(
                img1=a, img2=b, threshold=threshold, anti_spoofing=anti_spoofing
            ),
            ctx,
        )
        return res, ctx

    def analyze(self, img_b64: str, anti_spoofing: bool = False):
        ctx = _FakeContext()
        res = self.svc.FaceAnalyze(
            backend_pb2.FaceAnalyzeRequest(img=img_b64, anti_spoofing=anti_spoofing),
            ctx,
        )
        return res, ctx


class InsightFaceEngineTest(unittest.TestCase):