FIX-modin-project#2473: Add tests for ExactStr

Signed-off-by: Vasilij Litvinov <vasilij.n.litvinov@intel.com>
vnlitvinov · Nov 25, 2020 · e303706 · e303706
1 parent 586efe8
commit e303706
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/modin/config/test/test_envvars.py b/modin/config/test/test_envvars.py
@@ -14,7 +14,7 @@
 import os
 import pytest
 
-from modin.config.envvars import EnvironmentVariable, _check_vars
+from modin.config.envvars import EnvironmentVariable, _check_vars, ExactStr
 
 
 @pytest.fixture
@@ -25,9 +25,9 @@ def make_unknown_env():
     del os.environ[varname]
 
 
-@pytest.fixture
-def make_custom_envvar():
-    class CustomVar(EnvironmentVariable, type=str):
+@pytest.fixture(params=[str, ExactStr])
+def make_custom_envvar(request):
+    class CustomVar(EnvironmentVariable, type=request.param):
         """ custom var """
 
         default = 10
@@ -40,7 +40,7 @@ class CustomVar(EnvironmentVariable, type=str):
 @pytest.fixture
 def set_custom_envvar(make_custom_envvar):
     os.environ[make_custom_envvar.varname] = "  custom  "
-    yield "Custom"
+    yield "Custom" if make_custom_envvar.type is str else "  custom  "
     del os.environ[make_custom_envvar.varname]