Add identify test

sdk/face/azure-ai-vision-face/assets.json

@@ -2,5 +2,5 @@
   "AssetsRepo": "Azure/azure-sdk-assets",
   "AssetsRepoPrefixPath": "python",
   "TagPrefix": "python/face/azure-ai-vision-face",
-  "Tag": "python/face/azure-ai-vision-face_5f10bb94fa"
+  "Tag": "python/face/azure-ai-vision-face_0b4013000f"
 }

sdk/face/azure-ai-vision-face/tests/test_identify.py

@@ -0,0 +1,104 @@
+# coding: utf-8
+
+# -------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for
+# license information.
+# --------------------------------------------------------------------------
+from devtools_testutils import AzureRecordedTestCase, recorded_by_proxy
+
+from azure.ai.vision.face import FaceClient, FaceAdministrationClient
+from azure.ai.vision.face.models import (
+    FaceDetectionModel,
+    FaceRecognitionModel,
+)
+
+from preparers import FaceClientPreparer, FacePreparer, FaceAdministrationClientPreparer
+from _shared.constants import TestImages
+from _shared import helpers
+
+class TestIdentify(AzureRecordedTestCase):
+    test_images = [
+        TestImages.IMAGE_FAMILY_1_DAD_1,
+        TestImages.IMAGE_FAMILY_1_DAUGHTER_1,
+        TestImages.IMAGE_FAMILY_1_MOM_1,
+        TestImages.IMAGE_FAMILY_1_SON_1,
+    ]
+    group_id = "identify"
+
+    def _setup_faces(self, client: FaceClient):
+        face_ids = []
+        image_path = helpers.get_image_path(TestImages.IMAGE_IDENTIFICATION1)
+        result = client.detect(
+            helpers.read_file_content(image_path),
+            detection_model=FaceDetectionModel.DETECTION_03,
+            recognition_model=FaceRecognitionModel.RECOGNITION_04,
+            return_face_id=True
+        )
+        for face in result:
+            face_ids.append(face.face_id)
+        return face_ids
+
+    def _setup_group(self, operations):
+        operations.create(self.group_id, name=self.group_id, recognition_model=FaceRecognitionModel.RECOGNITION_04)
+
+        person_ids = []
+        for image in self.test_images:
+            result = operations.create_person(self.group_id, name="test_person")
+            assert result.person_id
+            person_ids.append(result.person_id)
+            image_path = helpers.get_image_path(image)
+            operations.add_face(
+                self.group_id,
+                result.person_id,
+                helpers.read_file_content(image_path),
+                detection_model=FaceDetectionModel.DETECTION_03
+            )
+
+        poller = operations.begin_train(self.group_id)
+        poller.result()
+
+        return person_ids
+
+    def _teardown_group(self, operations):
+        operations.delete(self.group_id)
+
+    @FacePreparer()
+    @FaceClientPreparer()
+    @FaceAdministrationClientPreparer()
+    @recorded_by_proxy
+    def test_identify_from_person_group(self, client: FaceClient, administration_client: FaceAdministrationClient):
+        face_ids = self._setup_faces(client)
+        self._setup_group(administration_client.person_group)
+
+        identify_result = client.identify_from_person_group(face_ids=face_ids, person_group_id=self.group_id)
+
+        assert len(identify_result) == len(face_ids)
+        for result in identify_result:
+            assert result.candidates is not None
+            assert result.face_id is not None
+            for candidate in result.candidates:
+                assert candidate.confidence is not None
+                assert candidate.person_id is not None
+
+        self._teardown_group(administration_client.person_group)
+
+    @FacePreparer()
+    @FaceClientPreparer()
+    @FaceAdministrationClientPreparer()
+    @recorded_by_proxy
+    def test_identify_from_large_person_group(self, client: FaceClient, administration_client: FaceAdministrationClient):
+        face_ids = self._setup_faces(client)
+        self._setup_group(administration_client.large_person_group)
+
+        identify_result = client.identify_from_large_person_group(face_ids=face_ids, large_person_group_id=self.group_id)
+
+        assert len(identify_result) == len(face_ids)
+        for result in identify_result:
+            assert result.candidates is not None
+            assert result.face_id is not None
+            for candidate in result.candidates:
+                assert candidate.confidence is not None
+                assert candidate.person_id is not None
+
+        self._teardown_group(administration_client.large_person_group)

sdk/face/azure-ai-vision-face/tests/test_identify_async.py

@@ -0,0 +1,104 @@
+# coding: utf-8
+
+# -------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for
+# license information.
+# --------------------------------------------------------------------------
+from devtools_testutils import AzureRecordedTestCase
+from devtools_testutils.aio import recorded_by_proxy_async
+
+from azure.ai.vision.face.aio import FaceClient, FaceAdministrationClient
+from azure.ai.vision.face.models import (
+    FaceDetectionModel,
+    FaceRecognitionModel,
+)
+
+from preparers import AsyncFaceClientPreparer, FacePreparer, AsyncFaceAdministrationClientPreparer
+from _shared.constants import TestImages
+from _shared import helpers
+
+class TestIdentify(AzureRecordedTestCase):
+    test_images = [
+        TestImages.IMAGE_FAMILY_1_DAD_1,
+        TestImages.IMAGE_FAMILY_1_DAUGHTER_1,
+        TestImages.IMAGE_FAMILY_1_MOM_1,
+        TestImages.IMAGE_FAMILY_1_SON_1,
+    ]
+    group_id = "identify"
+    async def _setup_faces(self, client: FaceClient):
+        face_ids = []
+        image_path = helpers.get_image_path(TestImages.IMAGE_IDENTIFICATION1)
+        result = await client.detect(
+            helpers.read_file_content(image_path),
+            detection_model=FaceDetectionModel.DETECTION_03,
+            recognition_model=FaceRecognitionModel.RECOGNITION_04,
+            return_face_id=True
+        )
+        for face in result:
+            face_ids.append(face.face_id)
+        return face_ids
+
+    async def _setup_group(self, operations):
+        await operations.create(self.group_id, name=self.group_id, recognition_model=FaceRecognitionModel.RECOGNITION_04)
+
+        person_ids = []
+        for image in self.test_images:
+            result = await operations.create_person(self.group_id, name="test_person")
+            assert result.person_id
+            person_ids.append(result.person_id)
+            image_path = helpers.get_image_path(image)
+            await operations.add_face(
+                self.group_id,
+                result.person_id,
+                helpers.read_file_content(image_path),
+                detection_model=FaceDetectionModel.DETECTION_03
+            )
+
+        poller = await operations.begin_train(self.group_id)
+        await poller.wait()
+
+        return person_ids
+
+    async def _teardown_group(self, operations):
+        await operations.delete(self.group_id)
+
+    @FacePreparer()
+    @AsyncFaceClientPreparer()
+    @AsyncFaceAdministrationClientPreparer()
+    @recorded_by_proxy_async
+    async def test_identify_from_person_group(self, client: FaceClient, administration_client: FaceAdministrationClient):
+        face_ids = await self._setup_faces(client)
+        await self._setup_group(administration_client.person_group)
+
+        identify_result = await client.identify_from_person_group(face_ids=face_ids, person_group_id=self.group_id)
+
+        assert len(identify_result) == len(face_ids)
+        for result in identify_result:
+            assert result.candidates is not None
+            assert result.face_id is not None
+            for candidate in result.candidates:
+                assert candidate.confidence is not None
+                assert candidate.person_id is not None
+
+        await self._teardown_group(administration_client.person_group)
+
+    @FacePreparer()
+    @AsyncFaceClientPreparer()
+    @AsyncFaceAdministrationClientPreparer()
+    @recorded_by_proxy_async
+    async def test_identify_from_large_person_group(self, client: FaceClient, administration_client: FaceAdministrationClient):
+        face_ids = await self._setup_faces(client)
+        await self._setup_group(administration_client.large_person_group)
+
+        identify_result = await client.identify_from_large_person_group(face_ids=face_ids, large_person_group_id=self.group_id)
+
+        assert len(identify_result) == len(face_ids)
+        for result in identify_result:
+            assert result.candidates is not None
+            assert result.face_id is not None
+            for candidate in result.candidates:
+                assert candidate.confidence is not None
+                assert candidate.person_id is not None
+
+        await self._teardown_group(administration_client.large_person_group)