Auto reset eerepr in tests and allow unnecessary resets

README.md

@@ -53,6 +53,7 @@ import ee
 import eerepr
 
 ee.Initialize()
+eerepr.initialize()
 
 display(ee.FeatureCollection("LARSE/GEDI/GEDI02_A_002_INDEX").limit(3))
 ```
@@ -65,5 +66,3 @@ display(ee.FeatureCollection("LARSE/GEDI/GEDI02_A_002_INDEX").limit(3))
 ## Caching
 
 `eerepr` uses caching to improve performance. Server data will only be requested once for each unique Earth Engine object, and all subsequent requests will be retrieved from the cache until the Jupyter session is restarted.
-
-When you import `eerepr`, it is automatically initialized with an unlimited cache size. You can manually set the number of unique objects to cache using `eerepr.initialize(max_cache_size=n)`. A value of `None` sets an unlimited cache while a value of `0` disables caching.

eerepr/repr.py

@@ -146,4 +146,5 @@ def initialize(max_cache_size: int | None = None) -> None:
 def reset():
     """Remove HTML repr methods added by eerepr to EE objects."""
     for cls in reprs_set:
-        delattr(cls, REPR_HTML)
+        if hasattr(cls, REPR_HTML):
+            delattr(cls, REPR_HTML)

tests/conftest.py

@@ -3,11 +3,20 @@
 import ee
 import pytest
 
+import eerepr
+
 
 def pytest_sessionstart(session):
     ee.Initialize()
 
 
+@pytest.fixture(autouse=True)
+def reset_eerepr():
+    """Reset eerepr after each test to avoid inconsistent state."""
+    yield
+    eerepr.reset()
+
+
 @pytest.fixture(scope="session")
 def original_datadir():
     """Set the pytest-regressions directory."""

tests/test_reprs.py

@@ -1,7 +1,7 @@
 import ee
 import pytest
 
-import eerepr  # noqa: F401
+import eerepr
 from eerepr.repr import _repr_html_