[style] Format with black

Author: ao-picterra

examples/forge/detect_on_project.py

@@ -1,6 +1,7 @@
 """
 Demonstrate running a detector on all images within a project
 """
+
 from picterra import APIClient
 
 # Set the PICTERRA_API_KEY environment variable to define your API key

examples/forge/raster_management.py

@@ -25,14 +25,13 @@
 local_raster_id = client.upload_raster("data/raster1.tif", name="A short-lived raster")
 print("Uploaded a second local raster=", local_raster_id)
 # Editing the image's band specification. See https://docs.picterra.ch/imagery/#Multispectral
-client.edit_raster(local_raster_id, multispectral_band_specification={
-    "ranges": [
-        [0, 128], [0, 128], [0, 128]
-    ],
-    "display_bands": [
-        {"type": "multiband", "name": "default", "bands": [2, 1, 0]}
-    ]
-})
+client.edit_raster(
+    local_raster_id,
+    multispectral_band_specification={
+        "ranges": [[0, 128], [0, 128], [0, 128]],
+        "display_bands": [{"type": "multiband", "name": "default", "bands": [2, 1, 0]}],
+    },
+)
 # Deleting the image
 client.delete_raster(local_raster_id)
 print("Deleted raster=", local_raster_id)

examples/tracer/plots_analysis.py

@@ -13,7 +13,7 @@
     "New analysis",
     ["plotid_1", "plotid_2", "plotid_3"],
     datetime.date.fromisoformat("2022-01-01"),
-    datetime.date.fromisoformat("2024-01-01")
+    datetime.date.fromisoformat("2024-01-01"),
 )
 url = client.get_plots_analysis(analysis_id, plots_group_id)["url"]
 

tests/test_base_client.py

@@ -65,7 +65,9 @@ def request_callback(request, uri, response_headers):
         time.sleep(2)
         return [200, response_headers, json.dumps([])]
 
-    httpretty.register_uri(httpretty.GET, detector_api_url("rasters/"), body=request_callback)
+    httpretty.register_uri(
+        httpretty.GET, detector_api_url("rasters/"), body=request_callback
+    )
     timeout = 1
     client = _client(monkeypatch, timeout=timeout)
     with pytest.raises(requests.exceptions.ConnectionError) as e:
@@ -79,7 +81,9 @@ def request_callback(request, uri, response_headers):
 
 @responses.activate
 def test_headers_api_key(monkeypatch):
-    _add_api_response(detector_api_url("detectors/"), responses.POST, json={"id": "foobar"})
+    _add_api_response(
+        detector_api_url("detectors/"), responses.POST, json={"id": "foobar"}
+    )
     client = _client(monkeypatch)
     client.create_detector()
     assert len(responses.calls) == 1
@@ -88,7 +92,9 @@ def test_headers_api_key(monkeypatch):
 
 @responses.activate
 def test_headers_user_agent_version(monkeypatch):
-    _add_api_response(detector_api_url("detectors/"), responses.POST, json={"id": "foobar"})
+    _add_api_response(
+        detector_api_url("detectors/"), responses.POST, json={"id": "foobar"}
+    )
     client = _client(monkeypatch)
     client.create_detector()
     assert len(responses.calls) == 1
@@ -99,8 +105,12 @@ def test_headers_user_agent_version(monkeypatch):
 
 @responses.activate
 def test_headers_user_agent_version__fallback(monkeypatch):
-    _add_api_response(detector_api_url("detectors/"), responses.POST, json={"id": "foobar"},)
-    monkeypatch.setattr(base_client, '_get_distr_name', lambda: 'foobar')
+    _add_api_response(
+        detector_api_url("detectors/"),
+        responses.POST,
+        json={"id": "foobar"},
+    )
+    monkeypatch.setattr(base_client, "_get_distr_name", lambda: "foobar")
     client = _client(monkeypatch)
     client.create_detector()
     assert len(responses.calls) == 1
@@ -126,7 +136,9 @@ def test_results_page():
         method=responses.GET,
         url="http://example.com/page/2",
         json={
-            "count": 1, "next": None, "previous": "http://example.com/page/1",
+            "count": 1,
+            "next": None,
+            "previous": "http://example.com/page/1",
             "results": ["three"],
         },
         status=200,

tests/test_forge_client.py

@@ -79,7 +79,9 @@ def add_mock_rasters_in_folder_list_response(folder_id):
     }
     qs = {"folder": folder_id, "page_number": "1"}
     _add_api_response(
-        detector_api_url("rasters/"), json=data, match=responses.matchers.query_param_matcher(qs)
+        detector_api_url("rasters/"),
+        json=data,
+        match=responses.matchers.query_param_matcher(qs),
     )
 
 
@@ -110,7 +112,9 @@ def add_mock_rasters_in_filtered_list_response(
     if has_layers is not None:
         qs["has_vector_layers"] = bool(has_layers)
     _add_api_response(
-        detector_api_url("rasters/"), match=responses.matchers.query_param_matcher(qs), json=data
+        detector_api_url("rasters/"),
+        match=responses.matchers.query_param_matcher(qs),
+        json=data,
     )
 
 
@@ -186,21 +190,32 @@ def add_mock_detectors_list_response(string=None, tag=None, shared=None):
 
 def add_mock_detector_creation_response(**kwargs):
     match = responses.json_params_matcher({"configuration": kwargs}) if kwargs else None
-    _add_api_response(detector_api_url("detectors/"), responses.POST, json={"id": "foobar"}, match=match)
+    _add_api_response(
+        detector_api_url("detectors/"),
+        responses.POST,
+        json={"id": "foobar"},
+        match=match,
+    )
 
 
 def add_mock_detector_edit_response(d_id, **kwargs):
     match = responses.json_params_matcher({"configuration": kwargs}) if kwargs else None
-    _add_api_response(detector_api_url("detectors/%s/" % d_id), responses.PUT, status=204, match=match)
+    _add_api_response(
+        detector_api_url("detectors/%s/" % d_id), responses.PUT, status=204, match=match
+    )
 
 
 def add_mock_detector_train_responses(detector_id):
-    _add_api_response(detector_api_url("detectors/%s/train/" % detector_id), responses.POST, OP_RESP)
+    _add_api_response(
+        detector_api_url("detectors/%s/train/" % detector_id), responses.POST, OP_RESP
+    )
 
 
 def add_mock_run_dataset_recommendation_responses(detector_id):
     _add_api_response(
-        detector_api_url("detectors/%s/dataset_recommendation/" % detector_id), responses.POST, OP_RESP
+        detector_api_url("detectors/%s/dataset_recommendation/" % detector_id),
+        responses.POST,
+        OP_RESP,
     )
 
 
@@ -287,7 +302,9 @@ def add_mock_raster_upload_responses(identity_key, multispectral, cloud_coverage
     # Storage PUT
     responses.add(responses.PUT, "http://storage.example.com", status=200)
     # Commit
-    _add_api_response(detector_api_url("rasters/%s/commit/" % raster_id), responses.POST, OP_RESP)
+    _add_api_response(
+        detector_api_url("rasters/%s/commit/" % raster_id), responses.POST, OP_RESP
+    )
     # Status, first check
     data = {"id": raster_id, "name": "raster1", "status": "processing"}
     _add_api_response(detector_api_url("rasters/%s/" % raster_id), json=data)
@@ -302,26 +319,36 @@ def add_mock_detection_areas_upload_responses(raster_id):
     # Upload initiation
     data = {"upload_url": "http://storage.example.com", "upload_id": upload_id}
     _add_api_response(
-        detector_api_url("rasters/%s/detection_areas/upload/file/" % raster_id), responses.POST, data
+        detector_api_url("rasters/%s/detection_areas/upload/file/" % raster_id),
+        responses.POST,
+        data,
     )
     # Storage PUT
     responses.add(responses.PUT, "http://storage.example.com", status=200)
     # Commit
     _add_api_response(
-        detector_api_url("rasters/%s/detection_areas/upload/%s/commit/" % (raster_id, upload_id)),
+        detector_api_url(
+            "rasters/%s/detection_areas/upload/%s/commit/" % (raster_id, upload_id)
+        ),
         responses.POST,
         OP_RESP,
         status=200,
     )
     # Status, first check
     data = {"status": "processing"}
     _add_api_response(
-        detector_api_url("rasters/%s/detection_areas/upload/%s/" % (raster_id, upload_id)), json=data
+        detector_api_url(
+            "rasters/%s/detection_areas/upload/%s/" % (raster_id, upload_id)
+        ),
+        json=data,
     )
     # Status, second check
     data = {"status": "ready"}
     _add_api_response(
-        detector_api_url("rasters/%s/detection_areas/upload/%s/" % (raster_id, upload_id)), json=data
+        detector_api_url(
+            "rasters/%s/detection_areas/upload/%s/" % (raster_id, upload_id)
+        ),
+        json=data,
     )
 
 
@@ -390,7 +417,9 @@ def add_mock_vector_layer_download_responses(layer_id, polygons_num):
     }
     add_mock_operations_responses("success", results=results)
     url = results["download_url"]
-    polygons_fc = multipolygon_to_polygon_feature_collection(make_geojson_multipolygon(polygons_num))
+    polygons_fc = multipolygon_to_polygon_feature_collection(
+        make_geojson_multipolygon(polygons_num)
+    )
     assert len(polygons_fc["features"]) == polygons_num
     responses.add(
         responses.GET,
@@ -399,13 +428,19 @@ def add_mock_vector_layer_download_responses(layer_id, polygons_num):
     )
     return polygons_fc
 
+
 def add_mock_folder_creation_response(id, name):
     match = responses.matchers.json_params_matcher({"name": name})
-    _add_api_response(detector_api_url("folders/"), responses.POST, json={"id": id}, match=match)
+    _add_api_response(
+        detector_api_url("folders/"), responses.POST, json={"id": id}, match=match
+    )
 
 
 def make_geojson_polygon(base=1):
-    return {"type": "Polygon", "coordinates": [[[0, 0], [base, 0], [base, base], [0, base], [0, 0]]]}
+    return {
+        "type": "Polygon",
+        "coordinates": [[[0, 0], [base, 0], [base, base], [0, base], [0, 0]]],
+    }
 
 
 def make_geojson_multipolygon(npolygons=1):
@@ -424,9 +459,10 @@ def add_mock_download_result_response(op_id, num_classes):
                     "class": {"name": f"class_{i + 1}"},
                     "result": {
                         "url": f"http://storage.example.com/result_for_class_{i + 1}.geojson",
-                        "vector_layer_id": f"layer_{i + 1}"
+                        "vector_layer_id": f"layer_{i + 1}",
                     },
-                } for i in range(num_classes)
+                }
+                for i in range(num_classes)
             ],
         },
     }
@@ -479,15 +515,19 @@ def add_mock_delete_raster_response(raster_id):
 
 
 def add_mock_delete_detectionarea_response(raster_id):
-    _add_api_response(detector_api_url("rasters/%s/detection_areas/" % raster_id), responses.DELETE)
+    _add_api_response(
+        detector_api_url("rasters/%s/detection_areas/" % raster_id), responses.DELETE
+    )
 
 
 def add_mock_delete_detector_response(detector_id):
     _add_api_response(detector_api_url("detectors/%s/" % detector_id), responses.DELETE)
 
 
 def add_mock_delete_vector_layer_response(layer_id):
-    _add_api_response(detector_api_url("vector_layers/%s/" % layer_id), responses.DELETE)
+    _add_api_response(
+        detector_api_url("vector_layers/%s/" % layer_id), responses.DELETE
+    )
 
 
 def add_mock_edit_vector_layer_response(layer_id, **kwargs):
@@ -536,7 +576,9 @@ def add_mock_marker_creation_response(marker_id, raster_id, detector_id, coords,
         "text": text,
     }
     match = responses.matchers.json_params_matcher(body)
-    _add_api_response(detector_api_url(url), responses.POST, json={"id": marker_id}, match=match)
+    _add_api_response(
+        detector_api_url(url), responses.POST, json={"id": marker_id}, match=match
+    )
 
 
 def add_mock_folder_detector_response(folder_id: str):
@@ -598,27 +640,30 @@ def test_multipolygon_to_polygon_feature_collection():
         "type": "MultiPolygon",
         "coordinates": [
             [[[0, 0], [0, 1], [1, 1], [1, 0], [0, 0]]],
-            [[[1, 1], [1, 2], [2, 2], [2, 1], [1, 1]]]
-        ]
+            [[[1, 1], [1, 2], [2, 2], [2, 1], [1, 1]]],
+        ],
     }
     fc = multipolygon_to_polygon_feature_collection(mp)
     assert fc == {
         "type": "FeatureCollection",
-        "features": [{
-            "type": "Feature",
-            "properties": {},
-            "geometry": {
-                "type": "Polygon",
-                "coordinates": [[[0, 0], [0, 1], [1, 1], [1, 0], [0, 0]]]
-            }
-        },  {
-            "type": "Feature",
-            "properties": {},
-            "geometry": {
-                "type": "Polygon",
-                "coordinates": [[[1, 1], [1, 2], [2, 2], [2, 1], [1, 1]]]
-            }
-        }]
+        "features": [
+            {
+                "type": "Feature",
+                "properties": {},
+                "geometry": {
+                    "type": "Polygon",
+                    "coordinates": [[[0, 0], [0, 1], [1, 1], [1, 0], [0, 0]]],
+                },
+            },
+            {
+                "type": "Feature",
+                "properties": {},
+                "geometry": {
+                    "type": "Polygon",
+                    "coordinates": [[[1, 1], [1, 2], [2, 2], [2, 1], [1, 1]]],
+                },
+            },
+        ],
     }
 
 
@@ -953,7 +998,9 @@ def test_download_result_to_feature_collection(monkeypatch):
         assert len(feat1["geometry"]["coordinates"]) == 10
         assert isinstance(feat1["geometry"]["coordinates"][0][0][0][0], (int, float))
         feat2 = fc["features"][(class_1_index + 1) % 2]
-        assert feat2["type"] == "Feature" and feat2["geometry"]["type"] == "MultiPolygon"
+        assert (
+            feat2["type"] == "Feature" and feat2["geometry"]["type"] == "MultiPolygon"
+        )
         assert feat2["properties"]["class_name"] == "class_2"
         assert len(feat2["geometry"]["coordinates"]) == 20
         assert isinstance(feat2["geometry"]["coordinates"][0][0][0][0], (int, float))
@@ -1060,7 +1107,9 @@ def test_download_vector_layer_to_file(monkeypatch):
     assert fc["type"] == "FeatureCollection"
     assert fc == polygons_fc and len(fc["features"]) == 2
     assert fc["features"][0]["geometry"]["type"] == "Polygon"
-    assert isinstance(fc["features"][1]["geometry"]["coordinates"][0][0][0], (int, float))
+    assert isinstance(
+        fc["features"][1]["geometry"]["coordinates"][0][0][0], (int, float)
+    )
     assert len(responses.calls) == 3  # POST /download, GET /operations, GET url