Fix bug in memory driver implementation.

opendatacube · Jan 24, 2024 · f41718f · f41718f
1 parent 74c5cfd
commit f41718f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/datacube/index/memory/_datasets.py b/datacube/index/memory/_datasets.py
@@ -654,7 +654,7 @@ def temporal_extent(
         elif product is not None:
             if isinstance(product, str):
                 product = self._index.products.get_by_name_unsafe(product)
-            ids = self.by_product.get(product, [])
+            ids = self.by_product.get(product.name, [])
 
         min_time: Optional[datetime.datetime] = None
         max_time: Optional[datetime.datetime] = None

diff --git a/integration_tests/index/test_memory_index.py b/integration_tests/index/test_memory_index.py
@@ -400,7 +400,8 @@ def test_temporal_extent(mem_eo3_data: ODCEnvironment):
         assert (tmin is None and tmax is None) or tmin < tmax
         tmin2, tmax2 = dc.index.datasets.temporal_extent(product=ds.product)
         assert tmin2 == tmin and tmax2 == tmax
-        tmin2, tmax2 = dc.index.datasets.temporal_extent(ids=[ds.id])
+        ids = [doc["id"] for prod, doc, uris in dc.index.datasets.get_all_docs_for_product(product=ds.product)]
+        tmin2, tmax2 = dc.index.datasets.temporal_extent(ids=ids)
         assert tmin2 == tmin and tmax2 == tmax