From 7b998cdaf6376f48500965b9bbf37ad9a4b79ea2 Mon Sep 17 00:00:00 2001 From: Matthias Bisping Date: Mon, 23 May 2022 13:14:18 +0200 Subject: [PATCH] refactoring --- test/fixtures/server.py | 23 ++++++++++++----------- 1 file changed, 12 insertions(+), 11 deletions(-) diff --git a/test/fixtures/server.py b/test/fixtures/server.py index 959cdf6..6c94fc2 100644 --- a/test/fixtures/server.py +++ b/test/fixtures/server.py @@ -95,18 +95,19 @@ def core_operation(item_type, one_to_many, analysis_task): # yield page.get_pixmap().tobytes("png"), metadata yield f"page_{i}".encode() + params2op = { + False: { + "string": {False: upper}, + "image": {False: rotate, True: classify}, + }, + True: { + "string": {False: duplicate}, + "pdf": {False: stream_pages}, + }, + } + try: - d = { - False: { - "string": {False: upper}, - "image": {False: rotate, True: classify}, - }, - True: { - "string": {False: duplicate}, - "pdf": {False: stream_pages}, - }, - } - return d[one_to_many][item_type][analysis_task] + return params2op[one_to_many][item_type][analysis_task] except KeyError: pytest.skip(f"No operation defined for parameter combination.") logger.debug(f"No operation defined for [{one_to_many=}, {item_type=}, {analysis_task=}].")