Adjust test, take ownership when :memory: has been passed

Author: sultaniman

dlt/destinations/impl/duckdb/configuration.py

@@ -72,12 +72,14 @@ def parse_native_representation(self, native_value: Any) -> None:
             # if in-memory mode has been requested
             # then prepare duckdb connection so later
             # it is configured appropriately
+            is_owner = False
             if isinstance(native_value, str) and native_value == ":memory:":
                 native_value = duckdb.connect(native_value)
+                is_owner = True
 
             if isinstance(native_value, duckdb.DuckDBPyConnection):
                 self._conn = native_value
-                self._conn_owner = False
+                self._conn_owner = is_owner
                 self._conn_borrows = 0
                 self.database = ":external:"
                 self.__is_resolved__ = True

tests/load/duckdb/test_duckdb_client.py

@@ -259,16 +259,13 @@ def test_external_duckdb_database() -> None:
 
 
 def test_in_memory_duckdb_database() -> None:
-    import duckdb
-
     # pass explicit in memory database
-    conn = duckdb.connect(":memory:")
     c = resolve_configuration(
-        DuckDbClientConfiguration(credentials=conn)._bind_dataset_name(dataset_name="test_dataset")
+        DuckDbClientConfiguration(credentials=":memory:")._bind_dataset_name(dataset_name="test_dataset")
     )
-    c.credentials.borrow_conn(read_only=False)
+    conn = c.credentials.borrow_conn(read_only=False)
     assert c.credentials._conn_borrows == 1
-    assert c.credentials._conn_owner is False
+    assert c.credentials._conn_owner
     conn.close()
     assert not os.path.exists(":memory:")