Use storage format variable in asserts

audeering · Jul 26, 2024 · a7c3062 · a7c3062
1 parent 7acf38c
commit a7c3062
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/tests/test_load_on_demand.py b/tests/test_load_on_demand.py
@@ -77,7 +77,7 @@ def dbs(tmpdir_factory, persistent_repository, storage_format):
     return paths
 
 
-def test_load_only_metadata(dbs):
+def test_load_only_metadata(dbs, storage_format):
     db_original = audformat.Database.load(dbs[DB_VERSION])
 
     db = audb.load(
@@ -103,7 +103,7 @@ def test_load_only_metadata(dbs):
 
     # Delete table1
     # to force downloading from backend again
-    os.remove(os.path.join(db.meta["audb"]["root"], "db.table1.csv"))
+    os.remove(os.path.join(db.meta["audb"]["root"], f"db.table1.{storage_format}"))
     os.remove(os.path.join(db.meta["audb"]["root"], "db.table1.pkl"))
     db = audb.load(
         DB_NAME,

diff --git a/tests/test_publish.py b/tests/test_publish.py
@@ -1085,7 +1085,7 @@ def test_publish_text_media_files(tmpdir, dbs, repository, storage_format):
     version = "1.0.0"
     deps = audb.publish(build_dir, version, repository)
 
-    assert deps.tables == ["db.files.csv"]
+    assert deps.tables == [f"db.files.{storage_format}"]
     file = "data/file1.txt"
     assert deps.media == [file]
     assert deps.bit_depth(file) == 0