Create test_rdd_not_installed.py

DoubleML · Jan 8, 2025 · ad4b6c8 · ad4b6c8
1 parent bcf62c2
commit ad4b6c8
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/doubleml/rdd/tests/test_rdd_not_installed.py b/doubleml/rdd/tests/test_rdd_not_installed.py
@@ -0,0 +1,11 @@
+import pytest
+from unittest.mock import patch
+
+
+@pytest.mark.ci
+def test_rdrobust_import_error():
+    with patch('doubleml.rdd.rdd._rdrobust_available', False):
+        msg = r"rdrobust is not installed. Please install it using 'pip install DoubleML\[rdd\]'"
+        with pytest.raises(ImportError, match=msg):
+            from doubleml.rdd import RDFlex
+            RDFlex(None, None)