[test] lambda_handler: add test rectangle and multi prompts cases

src/io/lambda_helpers.py

@@ -116,7 +116,7 @@ def _get_new_prompt_data_rectangle(bbox_ne, bbox_sw, prompt, zoom):
     ]
 
 
-def get_parsed_request_body(event: Dict) -> RawRequestInput:
+def get_parsed_request_body(event: Dict or str) -> RawRequestInput:
     """
     Validator for the raw input request lambda event
 

tests/events/get_parsed_bbox_prompts_multi_prompt.json

@@ -0,0 +1,109 @@
+{
+  "input": {
+    "bbox": {
+      "ne": {
+        "lat": 46.28788479194308,
+        "lng": 9.475707775999172
+      },
+      "sw": {
+        "lat": 46.12536667198419,
+        "lng": 9.124145275999172
+      }
+    },
+    "prompt": [
+      {
+        "id": 527,
+        "type": "point",
+        "data": {
+          "lat": 46.18361019327142,
+          "lng": 9.442899846498142
+        },
+        "label": 1
+      },
+      {
+        "id": 861,
+        "type": "rectangle",
+        "data": {
+          "ne": {
+            "lat": 46.19067068237379,
+            "lng": 9.448158649569828
+          },
+          "sw": {
+            "lat": 46.154770662283525,
+            "lng": 9.40386976089288
+          }
+        }
+      },
+      {
+        "id": 869,
+        "type": "point",
+        "data": {
+          "lat": 46.16404509756208,
+          "lng": 9.431335738366943
+        },
+        "label": 0
+      },
+      {
+        "id": 890,
+        "type": "point",
+        "data": {
+          "lat": 46.16975166534373,
+          "lng": 9.414512827164097
+        },
+        "label": 1
+      }
+    ],
+    "zoom": 12,
+    "source_type": "Satellite",
+    "url_tile": "http://localhost:8000/lambda_handler/{z}/{x}/{y}.png"
+  },
+  "output": {
+    "bbox": [
+      [
+        46.28788479194308,
+        9.475707775999172
+      ],
+      [
+        46.12536667198419,
+        9.124145275999172
+      ]
+    ],
+    "prompt": [
+      {
+        "type": "point",
+        "label": 1,
+        "data": [
+          929,
+          440
+        ]
+      },
+      {
+        "type": "rectangle",
+        "data": [
+          815,
+          410,
+          944,
+          561
+        ]
+      },
+      {
+        "type": "point",
+        "label": 0,
+        "data": [
+          895,
+          522
+        ]
+      },
+      {
+        "type": "point",
+        "label": 1,
+        "data": [
+          846,
+          498
+        ]
+      }
+    ],
+    "zoom": 12
+  }
+}
+

tests/events/get_parsed_bbox_prompts_single_rectangle.json

@@ -0,0 +1,58 @@
+{
+  "input": {
+    "bbox": {
+      "ne": {
+        "lat": -20.756702460787714,
+        "lng": 156.13847182385842
+      },
+      "sw": {
+        "lat": -45.55296601778451,
+        "lng": 111.13847182385842
+      }
+    },
+    "prompt": [
+      {
+        "id": 665,
+        "type": "rectangle",
+        "data": {
+          "ne": {
+            "lat": -39.71220032784001,
+            "lng": 149.33022207241362
+          },
+          "sw": {
+            "lat": -44.092362655875746,
+            "lng": 143.26567640979061
+          }
+        }
+      }
+    ],
+    "zoom": 5,
+    "source_type": "Satellite",
+    "url_tile": "http://localhost:8000/lambda_handler/{z}/{x}/{y}.png"
+  },
+  "output": {
+    "bbox": [
+      [
+        -20.756702460787714,
+        156.13847182385842
+      ],
+      [
+        -45.55296601778451,
+        111.13847182385842
+      ]
+    ],
+    "prompt": [
+      {
+        "data": [
+          731,
+          503,
+          869,
+          637
+        ],
+        "type": "rectangle"
+      }
+    ],
+    "zoom": 5
+  }
+}
+

tests/io/test_lambda_helpers.py

@@ -38,7 +38,7 @@ def test_get_response(time_mocked):
 
 
 def test_get_parsed_bbox_points():
-    with open(TEST_EVENTS_FOLDER / "get_parsed_bbox_points.json") as tst_json:
+    with open(TEST_EVENTS_FOLDER / "get_parsed_bbox_prompts_single_point.json") as tst_json:
         inputs_outputs = json.load(tst_json)
         for k, input_output in inputs_outputs.items():
             print(f"k:{k}.")
@@ -47,6 +47,15 @@ def test_get_parsed_bbox_points():
             assert output == input_output["output"]
 
 
+def test_get_parsed_bbox_other_inputs():
+    for json_filename in ["single_rectangle", "multi_prompt"]:
+        with open(TEST_EVENTS_FOLDER / f"get_parsed_bbox_prompts_{json_filename}.json") as tst_json:
+            inputs_outputs = json.load(tst_json)
+            parsed_input = RawRequestInput.model_validate(inputs_outputs["input"])
+            output = get_parsed_bbox_points(parsed_input)
+            assert output == inputs_outputs["output"]
+
+
 def test_get_parsed_request_body():
     input_event = {
         "event": {