arkindex-base-worker 0.4.0rc2__py3-none-any.whl → 0.4.0rc4__py3-none-any.whl

tests/test_image.py

@@ -1,6 +1,5 @@
 import logging
 import math
-import unittest
 import uuid
 from io import BytesIO
 from operator import attrgetter
@@ -219,74 +218,33 @@ def test_open_image_rotate_mirror(rotation_angle, mirrored, expected_path):
     assert _root_mean_square(expected, actual) <= 15.0
 
 
-class TestTrimPolygon(unittest.TestCase):
-    def test_trim_polygon_partially_outside_image(self):
-        bad_polygon = [
-            [99, 200],
-            [197, 224],
-            [120, 251],
-            [232, 350],
-            [312, 364],
-            [325, 310],
-            [318, 295],
-            [296, 260],
-            [352, 259],
-            [106, 210],
-            [197, 206],
-            [99, 200],
-        ]
-        expected_polygon = [
-            [99, 200],
-            [197, 224],
-            [120, 251],
-            [232, 300],
-            [312, 300],
-            [325, 300],
-            [318, 295],
-            [296, 260],
-            [352, 259],
-            [106, 210],
-            [197, 206],
-            [99, 200],
-        ]
-        assert (
-            trim_polygon(bad_polygon, TEST_IMAGE["width"], TEST_IMAGE["height"])
-            == expected_polygon
-        )
-
-    def test_trim_polygon_good_polygon(self):
-        good_polygon = (
-            (12, 56),
-            (29, 60),
-            (35, 61),
-            (42, 59),
-            (58, 57),
-            (65, 61),
-            (72, 57),
-            (12, 56),
-        )
-        expected_polygon = [
-            [12, 56],
-            [29, 60],
-            [35, 61],
-            [42, 59],
-            [58, 57],
-            [65, 61],
-            [72, 57],
-            [12, 56],
-        ]
-        assert (
-            trim_polygon(good_polygon, TEST_IMAGE["width"], TEST_IMAGE["height"])
-            == expected_polygon
-        )
-
-    def test_trim_polygon_invalid_polygon(self):
-        """
-        An assertion error is raised the polygon input isn't a list or tuple
-        """
-        bad_polygon = {
-            "polygon": [
-                [99, 200],
+@pytest.mark.parametrize(
+    ("polygon", "error"),
+    [
+        # Polygon input isn't a list or tuple
+        (
+            {
+                "polygon": [
+                    [99, 200],
+                    [25, 224],
+                    [0, 0],
+                    [0, 300],
+                    [102, 300],
+                    [260, 300],
+                    [288, 295],
+                    [296, 260],
+                    [352, 259],
+                    [106, 210],
+                    [197, 206],
+                    [99, 208],
+                ]
+            },
+            "Input polygon must be a valid list or tuple of points.",
+        ),
+        # Point coordinates are not integers
+        (
+            [
+                [9.9, 200],
                 [25, 224],
                 [0, 0],
                 [0, 300],
@@ -297,136 +255,161 @@ class TestTrimPolygon(unittest.TestCase):
                 [352, 259],
                 [106, 210],
                 [197, 206],
-                [99, 208],
-            ]
-        }
-        with pytest.raises(
-            AssertionError,
-            match="Input polygon must be a valid list or tuple of points.",
-        ):
-            trim_polygon(bad_polygon, TEST_IMAGE["width"], TEST_IMAGE["height"])
-
-    def test_trim_polygon_negative_coordinates(self):
-        """
-        Negative coordinates are ignored and replaced by 0 with no error being thrown
-        """
-        bad_polygon = [
-            [99, 200],
-            [25, 224],
-            [-8, -52],
-            [-12, 350],
-            [102, 364],
-            [260, 310],
-            [288, 295],
-            [296, 260],
-            [352, 259],
-            [106, 210],
-            [197, 206],
-            [99, 200],
-        ]
-        expected_polygon = [
-            [99, 200],
-            [25, 224],
-            [0, 0],
-            [0, 300],
-            [102, 300],
-            [260, 300],
-            [288, 295],
-            [296, 260],
-            [352, 259],
-            [106, 210],
-            [197, 206],
-            [99, 200],
-        ]
-        assert (
-            trim_polygon(bad_polygon, TEST_IMAGE["width"], TEST_IMAGE["height"])
-            == expected_polygon
-        )
+                [99, 20.8],
+            ],
+            "Polygon point coordinates must be integers.",
+        ),
+        # Point coordinates are not lists or tuples
+        (
+            [
+                [12, 56],
+                [29, 60],
+                [35, 61],
+                "[42, 59]",
+                [58, 57],
+                [65, 61],
+                [72, 57],
+                [12, 56],
+            ],
+            "Polygon points must be tuples or lists.",
+        ),
+        # Point coordinates are not lists or tuples of length 2
+        (
+            [
+                [12, 56],
+                [29, 60, 3],
+                [35, 61],
+                [42, 59],
+                [58, 57],
+                [65, 61],
+                [72, 57],
+                [12, 56],
+            ],
+            "Polygon points must be tuples or lists of 2 elements.",
+        ),
+        # None of the polygon's points are inside the image
+        (
+            [
+                [999, 200],
+                [1097, 224],
+                [1020, 251],
+                [1232, 350],
+                [1312, 364],
+                [1325, 310],
+                [1318, 295],
+                [1296, 260],
+                [1352, 259],
+                [1006, 210],
+                [997, 206],
+                [999, 200],
+            ],
+            "This polygon is entirely outside the image's bounds.",
+        ),
+    ],
+)
+def test_trim_polygon_errors(polygon, error):
+    with pytest.raises(AssertionError, match=error):
+        trim_polygon(polygon, TEST_IMAGE["width"], TEST_IMAGE["height"])
 
-    def test_trim_polygon_outside_image_error(self):
-        """
-        An assertion error is raised when none of the polygon's points are inside the image
-        """
-        bad_polygon = [
-            [999, 200],
-            [1097, 224],
-            [1020, 251],
-            [1232, 350],
-            [1312, 364],
-            [1325, 310],
-            [1318, 295],
-            [1296, 260],
-            [1352, 259],
-            [1006, 210],
-            [997, 206],
-            [999, 200],
-        ]
-        with pytest.raises(
-            AssertionError, match="This polygon is entirely outside the image's bounds."
-        ):
-            trim_polygon(bad_polygon, TEST_IMAGE["width"], TEST_IMAGE["height"])
-
-    def test_trim_polygon_float_coordinates(self):
-        """
-        An assertion error is raised when point coordinates are not integers
-        """
-        bad_polygon = [
-            [9.9, 200],
-            [25, 224],
-            [0, 0],
-            [0, 300],
-            [102, 300],
-            [260, 300],
-            [288, 295],
-            [296, 260],
-            [352, 259],
-            [106, 210],
-            [197, 206],
-            [99, 20.8],
-        ]
-        with pytest.raises(
-            AssertionError, match="Polygon point coordinates must be integers."
-        ):
-            trim_polygon(bad_polygon, TEST_IMAGE["width"], TEST_IMAGE["height"])
-
-    def test_trim_polygon_invalid_points_1(self):
-        """
-        An assertion error is raised when point coordinates are not lists or tuples
-        """
-        bad_polygon = [
-            [12, 56],
-            [29, 60],
-            [35, 61],
-            "[42, 59]",
-            [58, 57],
-            [65, 61],
-            [72, 57],
-            [12, 56],
-        ]
-        with pytest.raises(
-            AssertionError, match="Polygon points must be tuples or lists."
-        ):
-            trim_polygon(bad_polygon, TEST_IMAGE["width"], TEST_IMAGE["height"])
-
-    def test_trim_polygon_invalid_points_2(self):
-        """
-        An assertion error is raised when point coordinates are not lists or tuples of length 2
-        """
-        bad_polygon = [
-            [12, 56],
-            [29, 60, 3],
-            [35, 61],
-            [42, 59],
-            [58, 57],
-            [65, 61],
-            [72, 57],
-            [12, 56],
-        ]
-        with pytest.raises(
-            AssertionError,
-            match="Polygon points must be tuples or lists of 2 elements.",
-        ):
-            trim_polygon(bad_polygon, TEST_IMAGE["width"], TEST_IMAGE["height"])
+
+def test_trim_polygon_negative_coordinates():
+    """
+    Negative coordinates are ignored and replaced by 0 with no error being thrown
+    """
+    polygon = [
+        [99, 200],
+        [25, 224],
+        [-8, -52],
+        [-12, 350],
+        [102, 364],
+        [260, 310],
+        [288, 295],
+        [296, 260],
+        [352, 259],
+        [106, 210],
+        [197, 206],
+        [99, 200],
+    ]
+    expected_polygon = [
+        [99, 200],
+        [25, 224],
+        [0, 0],
+        [0, 300],
+        [102, 300],
+        [260, 300],
+        [288, 295],
+        [296, 260],
+        [352, 259],
+        [106, 210],
+        [197, 206],
+        [99, 200],
+    ]
+    assert (
+        trim_polygon(polygon, TEST_IMAGE["width"], TEST_IMAGE["height"])
+        == expected_polygon
+    )
+
+
+def test_trim_polygon_partially_outside_image():
+    polygon = [
+        [99, 200],
+        [197, 224],
+        [120, 251],
+        [232, 350],
+        [312, 364],
+        [325, 310],
+        [318, 295],
+        [296, 260],
+        [352, 259],
+        [106, 210],
+        [197, 206],
+        [99, 200],
+    ]
+    expected_polygon = [
+        [99, 200],
+        [197, 224],
+        [120, 251],
+        [232, 300],
+        [312, 300],
+        [325, 300],
+        [318, 295],
+        [296, 260],
+        [352, 259],
+        [106, 210],
+        [197, 206],
+        [99, 200],
+    ]
+    assert (
+        trim_polygon(polygon, TEST_IMAGE["width"], TEST_IMAGE["height"])
+        == expected_polygon
+    )
+
+
+def test_trim_polygon():
+    polygon = (
+        (12, 56),
+        (29, 60),
+        (35, 61),
+        (42, 59),
+        (58, 57),
+        (65, 61),
+        (72, 57),
+        (12, 56),
+    )
+    expected_polygon = [
+        [12, 56],
+        [29, 60],
+        [35, 61],
+        [42, 59],
+        [58, 57],
+        [65, 61],
+        [72, 57],
+        [12, 56],
+    ]
+    assert (
+        trim_polygon(polygon, TEST_IMAGE["width"], TEST_IMAGE["height"])
+        == expected_polygon
+    )
 
 
 @pytest.mark.parametrize(

arkindex_base_worker-0.4.0rc2.dist-info/RECORD

@@ -1,52 +0,0 @@
-arkindex_worker/__init__.py,sha256=OlgCtTC9MaWeejviY0a3iQpALcRQGMVArFVVYwTF6I8,162
-arkindex_worker/cache.py,sha256=FTlB0coXofn5zTNRTcVIvh709mcw4a1bPGqkwWjKs3w,11248
-arkindex_worker/image.py,sha256=oEgVCrSHiGh3D5-UXfM6PvT17TttSxC0115irpvB3Dw,18581
-arkindex_worker/models.py,sha256=bPQzGZNs5a6z6DEcygsa8T33VOqPlMUbwKzHqlKzwbw,9923
-arkindex_worker/utils.py,sha256=q1EeLdC6ebYIH-C0LOAqw2cNpjCjVoP-Vbr-39mF4w0,9884
-arkindex_worker/worker/__init__.py,sha256=4kj94AbdACO-q7w0krmv2QkCWbP7vLLgZk9sowZUBNw,15880
-arkindex_worker/worker/base.py,sha256=hIorYsLy88xR4-dhIsaQHbWCz5FMoja-gfRPCJhT3eY,19843
-arkindex_worker/worker/classification.py,sha256=ECm1cnQPOj_9m-CoO0e182ElSySAUOoyddHrORbShhc,10951
-arkindex_worker/worker/corpus.py,sha256=s9bCxOszJMwRq1WWAmKjWq888mjDfbaJ18Wo7h-rNOw,1827
-arkindex_worker/worker/dataset.py,sha256=LwzKwNFX4FqfLxh29LSvJydPwRw3VHaB1wjuFhUshsE,5267
-arkindex_worker/worker/element.py,sha256=Qvvq9kJnAHNATHW7zi96eIY1x-0MsR-T5rrSJg6e9Y4,45309
-arkindex_worker/worker/entity.py,sha256=qGjQvOVXfP84rER0Dkui6q-rb9nTWerHVG0Z5voB8pU,15229
-arkindex_worker/worker/image.py,sha256=t_Az6IGnj0EZyvcA4XxfPikOUjn_pztgsyxTkFZhaXU,621
-arkindex_worker/worker/metadata.py,sha256=VRajtd2kaBvar9GercX4knvR6l1WFYjoCdJWU9ccKgk,7291
-arkindex_worker/worker/process.py,sha256=I1rBt3Y8bV4zcPr8N1E2NRZ0UClSTqhExsO9CPcP41E,1012
-arkindex_worker/worker/task.py,sha256=1O9zrWXxe3na3TOcoHX5Pxn1875v7EU08BSsCPnb62g,1519
-arkindex_worker/worker/training.py,sha256=qnBFEk11JOWWPLTbjF-lZ9iFBdTPpQzZAzQ9a03J1j4,10874
-arkindex_worker/worker/transcription.py,sha256=52RY9kYsiR1sz9FxOigyo12Ker3VDbQ4U42gK9DpR3g,21146
-arkindex_worker/worker/version.py,sha256=JIT7OI3Mo7RPkNrjOB9hfqrsG-FYygz_zi4l8PbkuAo,1960
-hooks/pre_gen_project.py,sha256=xQJERv3vv9VzIqcBHI281eeWLWREXUF4mMw7PvJHHXM,269
-tests/__init__.py,sha256=DG--S6IpGl399rzSAjDdHL76CkOIeZIjajCcyUSDhOQ,241
-tests/conftest.py,sha256=2ocZ2x-mZQrNe9zvWwhWk2_4ExdaBHIB74SvtDlExRE,21580
-tests/test_base_worker.py,sha256=2EIYcd_3f9O0zB5WiGIQV0Cn9wndLvnEnSfcAE1qWWU,30607
-tests/test_cache.py,sha256=ii0gyr0DrG7ChEs7pmT8hMdSguAOAcCze4bRMiFQxuk,10640
-tests/test_dataset_worker.py,sha256=r5SWdXSWKXGBFAgH6ouF2Nk-k6lLJ6SgXjKYD2Vu_Es,22088
-tests/test_element.py,sha256=2G9M15TLxQRmvrWM9Kw2ucnElh4kSv_oF_5FYwwAxTY,13181
-tests/test_image.py,sha256=J3jqB5OhcdCpB6n0UnwivxrMlne8YjFLXhq1gBMANrs,26711
-tests/test_merge.py,sha256=TuOeUS0UCz66DPOQFFhc4NQBxIjZL9f5czi4XnvGrr4,8270
-tests/test_utils.py,sha256=_WJUPnt-pM_TQ0er4yjPZy-u_LePrHq1lxwk_teky7M,2544
-tests/test_elements_worker/__init__.py,sha256=Fh4nkbbyJSMv_VtjQxnWrOqTnxXaaWI8S9WU0VrzCHs,179
-tests/test_elements_worker/test_classifications.py,sha256=fXZ8cSzIWwZ6LHsY7tKsy9-Pp9fKyKUStIXS4ViBcek,27779
-tests/test_elements_worker/test_cli.py,sha256=a23i1pUDbXi23MUtbWwGEcLLrmc_YlrbDgOG3h66wLM,2620
-tests/test_elements_worker/test_corpus.py,sha256=c_LUHvkJIYgk_wXF06VQPNOoWfiZ06XpjOXrJ7MRiBc,4479
-tests/test_elements_worker/test_dataset.py,sha256=lSXqubhg1EEq2Y2goE8Y2RYaqIpM9Iejq6fGNW2BczU,11411
-tests/test_elements_worker/test_elements.py,sha256=JZmqddzxuEFhGNnYj8iE4kA7mODYbMGNf7STnA8CVL4,115496
-tests/test_elements_worker/test_entities.py,sha256=oav2dtvWWavQe1l3Drbxw1Ta2ocUJEVxJfDQ_r6-rYQ,36181
-tests/test_elements_worker/test_image.py,sha256=_E3UGdDOwTo1MW5KMS81PrdeSPBPWinWYoQPNy2F9Ro,2077
-tests/test_elements_worker/test_metadata.py,sha256=cm2NNaXxBYmYMkPexSPVTAqb2skDTB4mliwQCLz8Y98,22293
-tests/test_elements_worker/test_task.py,sha256=7Sr3fbjdgWUXJUhJEiC9CwnbhQIQX3rCInmHMIrmA38,5573
-tests/test_elements_worker/test_training.py,sha256=Qxi9EzGr_uKcn2Fh5aE6jNrq1K8QKLiOiSew4upASPs,8721
-tests/test_elements_worker/test_transcriptions.py,sha256=FNY6E26iTKqe7LP9LO72By4oV4g9hBIZYTU9BAc_w7I,77060
-tests/test_elements_worker/test_worker.py,sha256=AuFDyqncIusT-rMMY4sEay9MqGvoNuSuZQq-5rHN02U,10803
-worker-demo/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-worker-demo/tests/conftest.py,sha256=XzNMNeg6pmABUAH8jN6eZTlZSFGLYjS3-DTXjiRN6Yc,1002
-worker-demo/tests/test_worker.py,sha256=3DLd4NRK4bfyatG5P_PK4k9P9tJHx9XQq5_ryFEEFVg,304
-worker-demo/worker_demo/__init__.py,sha256=2BPomV8ZMNf3YXJgloatKeHQCE6QOkwmsHGkO6MkQuM,125
-worker-demo/worker_demo/worker.py,sha256=Rt-DjWa5iBP08k58NDZMfeyPuFbtNcbX6nc5jFX7GNo,440
-arkindex_base_worker-0.4.0rc2.dist-info/LICENSE,sha256=NVshRi1efwVezMfW7xXYLrdDr2Li1AfwfGOd5WuH1kQ,1063
-arkindex_base_worker-0.4.0rc2.dist-info/METADATA,sha256=Akdq43GSmQL2Sr7VpaBGMjKlQtb1z8lIEUUBqoXoWg8,3304
-arkindex_base_worker-0.4.0rc2.dist-info/WHEEL,sha256=UvcQYKBHoFqaQd6LKyqHw9fxEolWLQnlzP0h_LgJAfI,91
-arkindex_base_worker-0.4.0rc2.dist-info/top_level.txt,sha256=58NuslgxQC2vT4DiqZEgO4JqJRrYa2yeNI9QvkbfGQU,40
-arkindex_base_worker-0.4.0rc2.dist-info/RECORD,,