def test_as_string(self): leaf = ImportLeaf('') leaf.name, leaf.as_name = 'a', None self.assertEqual(leaf.as_string(), 'a') leaf.name, leaf.as_name = 'a', 'b' self.assertEqual(leaf.as_string(), 'a as b')
def test_str(self): leaf = ImportLeaf('a') self.assertEqual(six.text_type(leaf), leaf.as_string()) leaf = ImportLeaf('a as b') self.assertEqual(six.text_type(leaf), leaf.as_string()) leaf = ImportLeaf('a as a') self.assertEqual(six.text_type(leaf), leaf.as_string())