def test_from_import_as(self):
content = """from neutron.x import y1 as y
from neutron.w.x import y2 as z # comment
"""
lines = scanner.gen_parse(content.splitlines())
is_import, content, alias = next(lines)
self.assertTrue(is_import)
self.assertEqual('neutron.x.y1', content)
self.assertEqual('y', alias)
is_import, content, alias = next(lines)
self.assertTrue(is_import)
self.assertEqual('neutron.w.x.y2', content)
self.assertEqual('z', alias)
def test_import(self):
content = """import neutron.x.y1
import neutron.y2 # comment
"""
lines = scanner.gen_parse(content.splitlines())
is_import, content, alias = next(lines)
self.assertTrue(is_import)
self.assertEqual('neutron.x.y1', content)
self.assertEqual('y1', alias)
is_import, content, alias = next(lines)
self.assertTrue(is_import)
self.assertEqual('neutron.y2', content)
self.assertEqual('y2', alias)
def test_from_import_multiline(self):
content = """from neutron.x import (
y1)
from neutron.w.x import (
y2 )
"""
lines = scanner.gen_parse(content.splitlines())
is_import, content, alias = next(lines)
self.assertTrue(is_import)
self.assertEqual('neutron.x.y1', content)
self.assertEqual('y1', alias)
is_import, content, alias = next(lines)
self.assertTrue(is_import)
self.assertEqual('neutron.w.x.y2', content)
self.assertEqual('y2', alias)
def test_multiple_from_imports_on_one_line(self):
content = """from neutron.x import y1, y2, y3
"""
lines = scanner.gen_parse(content.splitlines())
is_import, content, alias = next(lines)
self.assertTrue(is_import)
self.assertEqual('neutron.x.y1', content)
self.assertEqual('y1', alias)
is_import, content, alias = next(lines)
self.assertTrue(is_import)
self.assertEqual('neutron.x.y2', content)
self.assertEqual('y2', alias)
is_import, content, alias = next(lines)
self.assertTrue(is_import)
self.assertEqual('neutron.x.y3', content)
self.assertEqual('y3', alias)
