Update database.py

Author: omad

cubedash/testutils/database.py

@@ -282,7 +282,7 @@ def odc_test_db(odc_db: str, cfg_env: ODCEnvironment) -> Generator[Datacube]:
 
             yield dc
 
-            dc.close()
+            
 
             # Drops the AGDC schema - badly named function
             pgres_core.drop_db(conn)
@@ -291,6 +291,8 @@ def odc_test_db(odc_db: str, cfg_env: ODCEnvironment) -> Generator[Datacube]:
             # and attempts to recreate them.
             _remove_postgres_dynamic_indexes()
 
+            dc.close()
+
     else:
         with index._db._engine.begin() as conn:  # type: ignore[attr-defined]
             for table in [
@@ -308,12 +310,11 @@ def odc_test_db(odc_db: str, cfg_env: ODCEnvironment) -> Generator[Datacube]:
 
             yield dc
 
-            dc.close()
-
             # Drop the ODC Schema - Badly named function
             pgis_core.drop_db(conn)
 
             _remove_postgis_dynamic_indexes()
+            dc.close()
 
 
 def _remove_postgres_dynamic_indexes() -> None: