test: fix tests

Author: dfguerrerom

pyproject.toml

@@ -51,7 +51,7 @@ dependencies = [
     "rio-tiler<7",
     "solara", 
     "ee-client>=1.1.1", # to handle multiple gee user sessions
-    "colorlog"
+    "colorlog",
     "reactivex"
 ]
 

sepal_ui/mapping/sepal_map.py

@@ -675,7 +675,7 @@ def get_viz_params(self, image: ee.Image) -> dict:
         Returns:
             The dictionary of the find properties
         """
-    
+
         # the constant prefix for SEPAL visualization parameters
         PREFIX = "visualization"
 
@@ -818,7 +818,7 @@ def add_basemap(self, basemap: str = "HYBRID") -> None:
             msg = f"Basemap can only be one of the following:\n{keys}"
             raise ValueError(msg)
 
-        self.add(eval(str(basemap_tiles[basemap])))
+        self.add_layer(eval(str(basemap_tiles[basemap])))
 
         return
 

sepal_ui/sepalwidgets/app.py

@@ -424,6 +424,11 @@ def vue_on_click_python(self, data=None) -> Self:
 
         return self
 
+    def _on_click(self, *args) -> Self:
+        """Display the appropriate tiles when the item is clicked."""
+        self.vue_on_click_python()
+        return self
+
 
 class NavDrawer(v.NavigationDrawer, SepalWidget):
     items: List[DrawerItem] = []

sepal_ui/sepalwidgets/inputs.py

@@ -30,7 +30,6 @@
 from typing_extensions import Self
 
 from sepal_ui.frontend import styles as ss
-from sepal_ui.logger.logger import logger
 from sepal_ui.message import ms
 from sepal_ui.scripts import decorator as sd
 from sepal_ui.scripts import utils as su
@@ -689,7 +688,6 @@ def __init__(
         """
         self._loaded = False
         self.valid = False
-        logger.debug(f"AssetSelect")
         self.gee_interface = GEEInterface(session=gee_session)
         # self.asset_info = {}
 

tests/test_mapping/test_SepalMap.py

@@ -208,7 +208,7 @@ def test_add_ee_layer(image_id: str) -> None:
     m = sm.SepalMap()
 
     # display all the viz available in the image
-    for viz in sm.SepalMap.get_viz_params(image).values():
+    for viz in sm.SepalMap().get_viz_params(image).values():
         m.addLayer(image, {}, viz["name"], viz_name=viz["name"])
 
     assert len(m.layers) == 6
@@ -271,7 +271,7 @@ def test_get_viz_params(image_id: str) -> None:
         image_id: the AssetId of the GEE image
     """
     image = ee.Image(image_id)
-    res = sm.SepalMap.get_viz_params(image)
+    res = sm.SepalMap().get_viz_params(image)
 
     expected = {
         "1": {
@@ -548,7 +548,7 @@ def ee_map_with_layers(image_id: str) -> sm.SepalMap:
     m = sm.SepalMap()
 
     # display all the viz available in the image
-    for viz in sm.SepalMap.get_viz_params(image).values():
+    for viz in sm.SepalMap().get_viz_params(image).values():
         m.addLayer(image, {}, viz["name"], viz_name=viz["name"])
 
     return m

tests/test_sepalwidgets/test_AssetSelect.py

@@ -1,7 +1,9 @@
 """Test the AssetSelect widget."""
 
+from datetime import time
 from pathlib import Path
 from typing import List
+import time
 
 import ee
 import pytest
@@ -11,15 +13,14 @@
 
 
 @pytest.mark.skipif(not ee.data._credentials, reason="GEE is not set")
-def test_init(gee_dir: Path, gee_user_dir: Path) -> None:
+def test_init(gee_dir: Path, gee_user_dir: Path, asset_select: sw.AssetSelect) -> None:
     """Init the widget.
 
     Args:
         gee_dir: the session defined GEE directory
         gee_user_dir: the gee_dir without the project information
     """
-    # create an asset select that points to the folder I created for testing
-    asset_select = sw.AssetSelect(folder=str(gee_dir))
+    # Test the fixture
     assert isinstance(asset_select, sw.AssetSelect)
     assert str(gee_user_dir / "image") in asset_select.items
 
@@ -156,12 +157,13 @@ def default_items() -> List[str]:
 
 @pytest.fixture(scope="function")
 def asset_select(gee_dir: Path) -> sw.AssetSelect:
-    """Create a default assetSelect.
-
-    Args:
-        gee_dir: the path to the session defined GEE directory
-
-    Returns:
-        The assertSelected wired to the session folder
-    """
-    return sw.AssetSelect(folder=str(gee_dir))
+    asset_select = sw.AssetSelect(folder=str(gee_dir))
+    # Wait for the asset select to finish loading
+    timeout = 5
+    start_time = time.time()
+    while not getattr(asset_select, "_loaded", False):
+        if time.time() - start_time > timeout:
+            raise TimeoutError("AssetSelect did not finish loading in time.")
+        time.sleep(0.1)
+
+    return asset_select