Function make_faces

python/example_code/rekognition/test/conftest.py:18–65

()

@pytest.fixture()
def make_faces():
    def _make_faces(count, has_details=False, is_celebrity=False, is_index=False):
        faces = []
        for _ in range(0, count):
            face_dict = {
                "BoundingBox": {"Left": 0, "Top": 0.5, "Width": 0.3, "Height": 0.7},
                "Confidence": random.randint(0, 100),
                "Landmarks": [{"Type": "test"}],
                "Pose": {"Roll": random.randint(0, 100)},
                "Quality": {"Brightness": random.randint(0, 100)},
            }
            if has_details:
                face_dict.update(
                    {
                        "AgeRange": {
                            "Low": random.randint(0, 30),
                            "High": random.randint(30, 100),
                        },
                        "Smile": {"Value": random.choice([True, False])},
                        "Eyeglasses": {"Value": random.choice([True, False])},
                        "Sunglasses": {"Value": random.choice([True, False])},
                        "Gender": {"Value": random.choice(["Male", "Female"])},
                        "Beard": {"Value": random.choice([True, False])},
                        "Mustache": {"Value": random.choice([True, False])},
                        "EyesOpen": {"Value": random.choice([True, False])},
                        "MouthOpen": {"Value": random.choice([True, False])},
                        "Emotions": [
                            {"Type": emotion, "Confidence": random.randint(0, 100)}
                            for emotion in ["BEMUSED", "CAUTIOUSLY OPTIMISTIC"]
                        ],
                    }
                )
            if is_index:
                face_dict.update({"FaceId": "test-face-id", "ImageId": "test-image-id"})
            if is_celebrity:
                celeb = {
                    "Urls": ["http://example.com"],
                    "Name": "test-name",
                    "Id": "test-celeb-id",
                    "MatchConfidence": random.randint(0, 100),
                    "Face": face_dict,
                }
                faces.append(celeb)
            else:
                faces.append(face_dict)
        return faces

    return _make_faces


@pytest.fixture()