adjusted param fixtures for serve test
This commit is contained in:
parent
8a2b60a8f5
commit
9a47388017
4
test/fixtures/server.py
vendored
4
test/fixtures/server.py
vendored
@ -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}
|
||||
|
||||
@ -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(
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user