diff --git a/test/fixtures/server.py b/test/fixtures/server.py index 658f04b..5ef3b04 100644 --- a/test/fixtures/server.py +++ b/test/fixtures/server.py @@ -82,7 +82,7 @@ def core_operation(item_type, one_to_many, analysis_task): yield upper(string, metadata), metadata def upper(string: bytes, metadata): - return string.decode().upper().encode() + return string.decode().upper().encode(), metadata def extract(string: bytes, metadata): for c in project(dict(enumerate(string.decode())), metadata["pages"]).values(): @@ -90,7 +90,7 @@ def core_operation(item_type, one_to_many, analysis_task): def rotate(im: bytes, metadata): im = Image.open(io.BytesIO(im)) - return image_to_bytes(im.rotate(90)) + return image_to_bytes(im.rotate(90)), metadata def classify(_: bytes, metadata): return b"", {"classification": 1} diff --git a/test/integration_tests/serve_test.py b/test/integration_tests/serve_test.py index 48500aa..df6f103 100644 --- a/test/integration_tests/serve_test.py +++ b/test/integration_tests/serve_test.py @@ -29,7 +29,7 @@ from test.utils.input import pair_data_with_queue_message @pytest.mark.parametrize( "analysis_task", [ - # False, + False, True, ], ) @@ -39,9 +39,9 @@ from test.utils.input import pair_data_with_queue_message @pytest.mark.parametrize( "item_type", [ - # "string", + "string", "image", - # "pdf", + "pdf", ], ) @pytest.mark.parametrize(