def test_replace_star_imports_unknown_module_1():
input = PythonBlock(dedent('''
from omgnonexistentmodule75085477 import *
''').lstrip())
output = replace_star_imports(input)
expected = PythonBlock(dedent('''
from omgnonexistentmodule75085477 import *
''').lstrip())
assert output == expected
def test_replace_star_imports_relative_1():
# Not implemented (semi-intentionally), but at least don't crash.
input = PythonBlock(dedent('''
from .x import *
''').lstrip(), filename="/foo/test_replace_star_imports_relative_1.py")
output = replace_star_imports(input)
expected = PythonBlock(dedent('''
from .x import *
''').lstrip(), filename="/foo/test_replace_star_imports_relative_1.py")
assert output == expected
def test_replace_star_imports_1():
m = types.ModuleType("fake_test_module_345489")
m.__all__ = ['f1', 'f2', 'f3', 'f4', 'f5']
sys.modules["fake_test_module_345489"] = m
input = PythonBlock(dedent('''
from mod1 import f1
from fake_test_module_345489 import *
from mod2 import f5
''').lstrip(), filename="/foo/test_replace_star_imports_1.py")
output = replace_star_imports(input)
expected = PythonBlock(dedent('''
from fake_test_module_345489 import f1, f2, f3, f4
from mod2 import f5
''').lstrip(), filename="/foo/test_replace_star_imports_1.py")
assert output == expected