Test relative as-import.

zopefoundation · Feb 3, 2017 · bd6135f · bd6135f
1 parent 65e7e19
commit bd6135f
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/src/RestrictedPython/tests/security_in_syntax26.py b/src/RestrictedPython/tests/security_in_syntax26.py
@@ -4,10 +4,6 @@
 
 
 
-def relative_import_as_bad_name():
-    from .x import y as _leading_underscore
-
-
 def except_as_bad_name():
     try:
         1 / 0

diff --git a/tests/test_transformer.py b/tests/test_transformer.py
@@ -1371,6 +1371,13 @@ def test_transformer__RestrictingNodeTransformer__visit_Import__8(compile):
     assert result.errors == (import_errmsg % '_m',)
 
 
+@pytest.mark.parametrize(*compile)
+def test_transformer__RestrictingNodeTransformer__visit_Import__9(compile):
+    """It denies relative from importing as something starting with `_`."""
+    result = compile('from .x import y as _leading_underscore')
+    assert result.errors == (import_errmsg % '_leading_underscore',)
+
+
 @pytest.mark.parametrize(*compile)
 def test_transformer__RestrictingNodeTransformer__visit_ClassDef(compile):
     result = compile('class Good: pass')