adjusted param fixtures for serve test

This commit is contained in:
Matthias Bisping 2022-06-03 13:48:33 +02:00
parent 8a2b60a8f5
commit 9a47388017
2 changed files with 5 additions and 5 deletions

View File

@ -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}

View File

@ -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(