Merge pull request #59 from kshivakumar/tc_ffix

.gitignore

@@ -43,6 +43,7 @@ coverage.xml
 .project
 .pydevproject
 .idea
+.vscode
 
 # Rope
 .ropeproject

pdir/api.py

@@ -20,7 +20,7 @@
 from . import format
 
 if platform.system() == 'Windows':
-    from colorama import init
+    from colorama import init  # type: ignore
 
     init()  # To support Windows.
 

pyproject.toml

@@ -53,3 +53,7 @@ build-backend = "pdm.pep517.api"
 
 [tool.black]
 skip-string-normalization = true
+
+[[tool.mypy.overrides]]
+module = ["pytest"]
+ignore_missing_imports = true

tests/test_user_config.py

@@ -12,14 +12,6 @@
 
 @pytest.fixture
 def clean():
-    DEFAULT_CONFIG_FILE = os.path.join(os.path.expanduser('~'), '.pdir2config')
-    yield DEFAULT_CONFIG_FILE
-    if 'PDIR2_CONFIG_FILE' in os.environ:
-        if os.path.exists(os.environ['PDIR2_CONFIG_FILE']):
-            os.remove(os.environ['PDIR2_CONFIG_FILE'])
-        del os.environ['PDIR2_CONFIG_FILE']
-    if os.path.exists(DEFAULT_CONFIG_FILE):
-        os.remove(DEFAULT_CONFIG_FILE)
     for module in (
         'pdir',
         'pdir.api',
@@ -33,6 +25,15 @@ def clean():
             del modules[module]
         except KeyError:
             pass
+
+    DEFAULT_CONFIG_FILE = os.path.join(os.path.expanduser('~'), '.pdir2config')
+    yield DEFAULT_CONFIG_FILE
+    if 'PDIR2_CONFIG_FILE' in os.environ:
+        if os.path.exists(os.environ['PDIR2_CONFIG_FILE']):
+            os.remove(os.environ['PDIR2_CONFIG_FILE'])
+        del os.environ['PDIR2_CONFIG_FILE']
+    if os.path.exists(DEFAULT_CONFIG_FILE):
+        os.remove(DEFAULT_CONFIG_FILE)
 
 
 def test_default_env_without_config(clean):