Exemplos de remove_duplicates em Python

Exemplo n.º 1

    def test_remove_duplicates_normal_and_from(self):
        src = """
import a.b
from a import b
"""
        tree = ast.parse(src)
        self.assertFalse(import_utils.remove_duplicates(tree))
        self.assertEqual(len(tree.body), 2)

Exemplo n.º 2

    def test_remove_duplicates_aliases(self):
        src = """
import a
import a as ax
import a as ax2
import a as ax
"""
        tree = ast.parse(src)
        self.assertTrue(import_utils.remove_duplicates(tree))
        self.assertEqual(len(tree.body), 3)
        self.assertEqual(tree.body[0].names[0].asname, None)
        self.assertEqual(tree.body[1].names[0].asname, 'ax')
        self.assertEqual(tree.body[2].names[0].asname, 'ax2')

Exemplo n.º 3

    def test_remove_duplicates_empty_node(self):
        src = """
import a, b, c
import b, c
"""
        tree = ast.parse(src)
        self.assertTrue(import_utils.remove_duplicates(tree))

        self.assertEqual(len(tree.body), 1)
        self.assertEqual(len(tree.body[0].names), 3)
        self.assertEqual(tree.body[0].names[0].name, 'a')
        self.assertEqual(tree.body[0].names[1].name, 'b')
        self.assertEqual(tree.body[0].names[2].name, 'c')

Exemplo n.º 4

    def test_remove_duplicates(self):
        src = """
import a
import b
import c
import b
import d
"""
        tree = ast.parse(src)
        self.assertTrue(import_utils.remove_duplicates(tree))

        self.assertEqual(len(tree.body), 4)
        self.assertEqual(tree.body[0].names[0].name, 'a')
        self.assertEqual(tree.body[1].names[0].name, 'b')
        self.assertEqual(tree.body[2].names[0].name, 'c')
        self.assertEqual(tree.body[3].names[0].name, 'd')

Exemplo n.º 5

    def test_remove_duplicates_multiple(self):
        src = """
import a, b
import b, c
import d, a, e, f
"""
        tree = ast.parse(src)
        self.assertTrue(import_utils.remove_duplicates(tree))

        self.assertEqual(len(tree.body), 3)
        self.assertEqual(len(tree.body[0].names), 2)
        self.assertEqual(tree.body[0].names[0].name, 'a')
        self.assertEqual(tree.body[0].names[1].name, 'b')
        self.assertEqual(len(tree.body[1].names), 1)
        self.assertEqual(tree.body[1].names[0].name, 'c')
        self.assertEqual(len(tree.body[2].names), 3)
        self.assertEqual(tree.body[2].names[0].name, 'd')
        self.assertEqual(tree.body[2].names[1].name, 'e')
        self.assertEqual(tree.body[2].names[2].name, 'f')