Fix problems introduced by LFB unit test (open-mmlab#677)

* remove lmdb * use temporary directory
Naoki-Wake · Mar 5, 2021 · 0326abf · 0326abf
1 parent 7d52ab7
commit 0326abf
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 2 deletions.
diff --git a/.gitignore b/.gitignore
@@ -120,6 +120,9 @@ work_dirs/
 # Profile
 *.prof
 
+# lmdb
+*.mdb
+
 # unignore some data file in tests/data
 !tests/data/**/*.pkl
 !tests/data/**/*.pkl.json

diff --git a/tests/data/lfb/lmdb/data.mdb b/tests/data/lfb/lmdb/data.mdb
diff --git a/tests/data/lfb/lmdb/lock.mdb b/tests/data/lfb/lmdb/lock.mdb
diff --git a/tests/test_models/test_head.py b/tests/test_models/test_head.py
@@ -1,4 +1,5 @@
 import os.path as osp
+import tempfile
 from unittest.mock import Mock, patch
 
 import numpy as np
@@ -312,8 +313,9 @@ def test_tsm_head():
 def test_lfb_infer_head():
     """Test layer construction, attributes and forward function in lfb infer
     head."""
-    lfb_infer_head = LFBInferHead(
-        lfb_prefix_path='temp_dir', use_half_precision=True)
+    with tempfile.TemporaryDirectory() as tmpdir:
+        lfb_infer_head = LFBInferHead(
+            lfb_prefix_path=tmpdir, use_half_precision=True)
     lfb_infer_head.init_weights()
 
     st_feat_shape = (3, 16, 1, 8, 8)