Refine test

audeering · Apr 3, 2024 · 5a24d34 · 5a24d34
1 parent 1fe0faf
commit 5a24d34
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -1,5 +1,4 @@
 import os
-import tempfile
 import typing
 
 import numpy as np
@@ -21,7 +20,6 @@
         "rendered_templates",
     )
 )
-TMP = tempfile.gettempdir()
 
 
 @pytest.fixture
@@ -80,7 +78,8 @@ def bare_db(
     # Publish and load database
     audb.publish(db_path, pytest.VERSION, repository)
     db = audb.load(name, version=pytest.VERSION, verbose=False)
-    assert db.root.startswith(TMP)
+    tmp_root = str(tmpdir.parts()[1])
+    assert db.root.startswith(tmp_root)
     return db
 
 
@@ -132,7 +131,8 @@ def minimal_db(
     # Publish and load database
     audb.publish(db_path, pytest.VERSION, repository)
     db = audb.load(name, version=pytest.VERSION, verbose=False)
-    assert db.root.startswith(TMP)
+    tmp_root = str(tmpdir.parts()[1])
+    assert db.root.startswith(tmp_root)
     return db
 
 
@@ -226,7 +226,8 @@ def medium_db(
     # Publish and load database
     audb.publish(db_path, pytest.VERSION, repository)
     db = audb.load(name, version=pytest.VERSION, verbose=False)
-    assert db.root.startswith(TMP)
+    tmp_root = str(tmpdir.parts()[1])
+    assert db.root.startswith(tmp_root)
     return db