From 7866d5bae76e54c75ba71e49327660fc5856c10b Mon Sep 17 00:00:00 2001 From: ric-evans Date: Tue, 28 Jan 2025 17:21:35 -0600 Subject: [PATCH] tests: `cluster` typing --- tests/integration/test_rest_routes.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/tests/integration/test_rest_routes.py b/tests/integration/test_rest_routes.py index 51f88798..13a651d5 100644 --- a/tests/integration/test_rest_routes.py +++ b/tests/integration/test_rest_routes.py @@ -171,7 +171,11 @@ async def _assert_db_scanrequests_coll( classifiers=post_scan_body["classifiers"], # # cluster (condor) config - request_clusters=list([k, v] for k, v in post_scan_body["cluster"].items()), + request_clusters=( + list([k, v] for k, v in post_scan_body["cluster"].items()) + if isinstance(post_scan_body["cluster"], dict) + else post_scan_body["cluster"] + ), worker_memory_bytes=humanfriendly.parse_size("8GB"), worker_disk_bytes=humanfriendly.parse_size("1GB"), max_pixel_reco_time=post_scan_body["max_pixel_reco_time"],