def test_flatten_getattr(self): """ Flattening a :class:`ast.Getattr` returns a module.name """ self.assertEqual(flatten_name(ast.Getattr(ast.Name('mod'), 'attr')), 'mod.attr')
def test_flatten_nested_getattr(self): """ Flattening a nested :class:`ast.Getattr` returns a module.module...name """ nested = ast.Getattr(ast.Getattr(ast.Name('mod1'), 'mod2'), 'attr') self.assertEqual(flatten_name(nested), 'mod1.mod2.attr')
def test_base_case(self): """ Flattening just a :class:`ast.Name` returns the text """ self.assertEqual(flatten_name(ast.Name('hey')), 'hey')