Skip to content

Commit e59dee2

Browse files
committed
Add a separate test for in-memory mode when using duckdb
1 parent 495801c commit e59dee2

File tree

1 file changed

+13
-0
lines changed

1 file changed

+13
-0
lines changed

tests/load/duckdb/test_duckdb_client.py

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -258,6 +258,19 @@ def test_external_duckdb_database() -> None:
258258
assert not os.path.exists(":memory:")
259259

260260

261+
def test_in_memory_duckdb_database() -> None:
262+
import duckdb
263+
264+
# pass explicit in memory database
265+
conn = duckdb.connect(":memory:")
266+
c = resolve_configuration(DuckDbClientConfiguration(credentials=conn))
267+
c.credentials.borrow_conn(read_only=False)
268+
assert c.credentials._conn_borrows == 1
269+
assert c.credentials._conn_owner is False
270+
conn.close()
271+
assert not os.path.exists(":memory:")
272+
273+
261274
def test_default_duckdb_dataset_name() -> None:
262275
# Check if dataset_name does not collide with pipeline_name
263276
data = ["a", "b", "c"]

0 commit comments

Comments
 (0)