test: use import_module instead of pure import

jitingxu1 · Sep 13, 2024 · 5f88d12 · 5f88d12
1 parent 9e80c6a
commit 5f88d12
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/test_optional_dependencies.py b/tests/test_optional_dependencies.py
@@ -1,12 +1,13 @@
 import sys
+from importlib import import_module
 from unittest.mock import patch
 
 import pytest
 
 
 # https://stackoverflow.com/a/65034142/1093967
-@pytest.mark.parametrize("optional_dependency", ["pandas", "numpy", "pyarrow"])
+@pytest.mark.parametrize("optional_dependency", ["numpy", "pandas", "pyarrow"])
 def test_without_dependency(optional_dependency):
     with patch.dict(sys.modules, {optional_dependency: None}):
         assert "ibis_ml" not in sys.modules
-        import ibis_ml  # noqa: F401
+        import_module("ibis_ml")