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)