Fix - Correções e Testes atualizados

gabriel-russo · Aug 23, 2023 · 33dbbf0 · 33dbbf0
1 parent 890f3b6
commit 33dbbf0
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 5 deletions.
diff --git a/src/cbers4asat/cbers4asat.py b/src/cbers4asat/cbers4asat.py
@@ -7,7 +7,6 @@
 from os.path import isdir, join
 from os import getcwd, cpu_count, mkdir
 from geopandas import GeoDataFrame
-from pandas import json_normalize
 from shapely.geometry import Polygon
 
 
@@ -265,7 +264,9 @@ def to_geodataframe(products: Dict, crs: str = "EPSG:4326"):
             item_metadata.update(bbox=item.bbox)
             item_metadata.update(collection=item.collection)
             item_metadata.update(thumbnail=item.thumbnail)
-            item_metadata.update(urls=[{asset: item.url(asset)} for asset in item.assets])
+            item_metadata.update(
+                urls=[{asset: item.url(asset)} for asset in item.assets]
+            )
 
             for index, feature in enumerate(products.get("features")):
                 if feature["id"] == item.id:

diff --git a/tests/test_Cbers4aAPI.py b/tests/test_Cbers4aAPI.py
@@ -39,7 +39,8 @@ def json():
                     },
                     "bbox": [-48.3106, -16.4178, -47.2492, -15.3637],
                     "properties": {"x": "XYZ"},
-                    "assets": {"blue": {"href": "http://test.dev/image.tif"}},
+                    "assets": {"blue": {"href": "http://test.dev/image.tif"},
+                               "thumbnail": {"href": "http://test.dev/thumbnail.png"}},
                     "links": [{"x": "y"}],
                 }
             ],
@@ -77,7 +78,8 @@ def json():
             },
             "bbox": [-48.3106, -16.4178, -47.2492, -15.3637],
             "properties": {"x": "XYZ"},
-            "assets": {"blue": {"href": "http://test.dev/image.tif"}},
+            "assets": {"blue": {"href": "http://test.dev/image.tif"},
+                       "thumbnail": {"href": "http://test.dev/thumbnail.png"}},
             "links": [{"x": "y"}],
         }
 
@@ -105,7 +107,8 @@ class TestCbers4aAPI:
                 },
                 "bbox": [-48.3106, -16.4178, -47.2492, -15.3637],
                 "properties": {"x": "XYZ"},
-                "assets": {"blue": {"href": "http://test.dev/image.tif"}},
+                "assets": {"blue": {"href": "http://test.dev/image.tif"},
+                           "thumbnail": {"href": "http://test.dev/thumbnail.png"}},
             }
         ],
     }