Exemple #1
0
 def setUp(self):
     self.fiximports = FixImports()
     self.fiximports.printErrorMsg = lambda *_: None
Exemple #2
0
class TestFixImports(unittest.TestCase):
    def setUp(self):
        self.fiximports = FixImports()
        self.fiximports.printErrorMsg = lambda *_: None

    def oneTest(self, src, expected):
        src = dedent(src)
        expected = dedent(expected)

        res, content = self.fiximports.sortImportGroups("testedfn", src)
        self.assertEqual(content, expected)

    def testBasic(self):
        self.oneTest(
            """
            from twisted.internet import defer
            from twisted.trial import unittest
            from fiximports import FixImports
            """, """
            from fiximports import FixImports
            from twisted.internet import defer
            from twisted.trial import unittest
            """)

    def testGroups(self):
        self.oneTest(
            """
            from twisted.internet import defer
            from twisted.trial import unittest

            from fiximports import FixImports
            """, """
            from twisted.internet import defer
            from twisted.trial import unittest

            from fiximports import FixImports
            """)

    def testMixImports(self):
        self.oneTest(
            """
            from twisted.internet import defer
            from twisted.trial import unittest
            import fiximports
            """, """
            import fiximports

            from twisted.internet import defer
            from twisted.trial import unittest
            """)

    def testDontSort(self):
        self.oneTest(
            """
            from twisted.internet import defer,\\
                 reactor
            from twisted.trial import unittest
            import fiximports
            """, """
            from twisted.internet import defer,\\
                 reactor
            from twisted.trial import unittest
            import fiximports
            """)

    def testFutureFirst(self):
        self.oneTest(
            """
            from __future__ import with_statement
            from twisted.internet import defer
            from twisted.trial import unittest
            import fiximports
            """, """
            from __future__ import with_statement

            import fiximports

            from twisted.internet import defer
            from twisted.trial import unittest
            """)
Exemple #3
0
 def setUp(self):
     self.fiximports = FixImports()
     self.fiximports.printErrorMsg = lambda *_: None
Exemple #4
0
class TestFixImports(unittest.TestCase):

    def setUp(self):
        self.fiximports = FixImports()
        self.fiximports.printErrorMsg = lambda *_: None

    def oneTest(self, src, expected):
        src = dedent(src)
        expected = dedent(expected)

        res, content = self.fiximports.sortImportGroups("testedfn", src)
        self.assertEqual(content, expected)

    def testBasic(self):
        self.oneTest("""
            from twisted.internet import defer
            from twisted.trial import unittest
            from fiximports import FixImports
            """, """
            from fiximports import FixImports
            from twisted.internet import defer
            from twisted.trial import unittest
            """)

    def testGroups(self):
        self.oneTest("""
            from twisted.internet import defer
            from twisted.trial import unittest

            from fiximports import FixImports
            """, """
            from twisted.internet import defer
            from twisted.trial import unittest

            from fiximports import FixImports
            """)

    def testMixImports(self):
        self.oneTest("""
            from twisted.internet import defer
            from twisted.trial import unittest
            import fiximports
            """, """
            import fiximports

            from twisted.internet import defer
            from twisted.trial import unittest
            """)

    def testDontSort(self):
        self.oneTest("""
            from twisted.internet import defer,\\
                 reactor
            from twisted.trial import unittest
            import fiximports
            """, """
            from twisted.internet import defer,\\
                 reactor
            from twisted.trial import unittest
            import fiximports
            """)

    def testFutureFirst(self):
        self.oneTest("""
            from __future__ import with_statement
            from twisted.internet import defer
            from twisted.trial import unittest
            import fiximports
            """, """
            from __future__ import with_statement

            import fiximports

            from twisted.internet import defer
            from twisted.trial import unittest
            """)
Exemple #5
0
class TestFixImports(unittest.TestCase):
    def setUp(self):
        self.fiximports = FixImports()
        self.fiximports.printErrorMsg = lambda *_: None

    def oneTest(self, src, expected):
        src = dedent(src)
        expected = dedent(expected)

        res, content = self.fiximports.sortImportGroups("testedfn", src)
        self.assertEqual(content, expected)

    def testBasic(self):
        self.oneTest(
            """
            from twisted.internet import defer
            from twisted.trial import unittest
            from fiximports import FixImports
            """, """
            from fiximports import FixImports
            from twisted.internet import defer
            from twisted.trial import unittest
            """)

    def testGroups(self):
        self.oneTest(
            """
            from twisted.internet import defer
            from twisted.trial import unittest

            from fiximports import FixImports
            """, """
            from twisted.internet import defer
            from twisted.trial import unittest

            from fiximports import FixImports
            """)

    def testMixImports(self):
        self.oneTest(
            """
            from twisted.internet import defer
            from twisted.trial import unittest
            import fiximports
            """, """
            import fiximports

            from twisted.internet import defer
            from twisted.trial import unittest
            """)

    def testFutureFirst(self):
        self.oneTest(
            """
            from __future__ import with_statement
            from twisted.internet import defer
            from twisted.trial import unittest
            import fiximports
            """, """
            from __future__ import with_statement

            import fiximports

            from twisted.internet import defer
            from twisted.trial import unittest
            """)

    def testSplitBackslashedSingle(self):
        self.oneTest(
            """
            from twisted.internet import \\
                 reactor
            """, """
            from twisted.internet import reactor
            """)

    def testSplitBackslashedList(self):
        self.oneTest(
            """
            from twisted.internet import defer,\\
                 reactor
            from X import A, B, \\
                 C, D, E
            """, """
            from X import A
            from X import B
            from X import C
            from X import D
            from X import E
            from twisted.internet import defer
            from twisted.internet import reactor
            """)

    def testSplitBackslashedListMultiline(self):
        self.oneTest(
            """
            from X import A, B, \\
                 C, D, \\
                 E
            """, """
            from X import A
            from X import B
            from X import C
            from X import D
            from X import E
            """)
class TestFixImports(unittest.TestCase):

    def setUp(self):
        self.fiximports = FixImports()
        self.fiximports.printErrorMsg = lambda *_: None

    def oneTest(self, src, expected):
        src = dedent(src)
        expected = dedent(expected)

        res, content = self.fiximports.sortImportGroups("testedfn", src)
        self.assertEqual(content, expected)

    def testBasic(self):
        self.oneTest("""
            from twisted.internet import defer
            from twisted.trial import unittest
            from fiximports import FixImports
            """, """
            from fiximports import FixImports
            from twisted.internet import defer
            from twisted.trial import unittest
            """)

    def testGroups(self):
        self.oneTest("""
            from twisted.internet import defer
            from twisted.trial import unittest

            from fiximports import FixImports
            """, """
            from twisted.internet import defer
            from twisted.trial import unittest

            from fiximports import FixImports
            """)

    def testMixImports(self):
        self.oneTest("""
            from twisted.internet import defer
            from twisted.trial import unittest
            import fiximports
            """, """
            import fiximports

            from twisted.internet import defer
            from twisted.trial import unittest
            """)

    def testFutureFirst(self):
        self.oneTest("""
            from __future__ import with_statement
            from twisted.internet import defer
            from twisted.trial import unittest
            import fiximports
            """, """
            from __future__ import with_statement

            import fiximports

            from twisted.internet import defer
            from twisted.trial import unittest
            """)

    def testSplitBackslashedSingle(self):
        self.oneTest("""
            from twisted.internet import \\
                 reactor
            """, """
            from twisted.internet import reactor
            """)

    def testSplitBackslashedList(self):
        self.oneTest("""
            from twisted.internet import defer,\\
                 reactor
            from X import A, B, \\
                 C, D, E
            """, """
            from X import A
            from X import B
            from X import C
            from X import D
            from X import E
            from twisted.internet import defer
            from twisted.internet import reactor
            """)

    def testSplitBackslashedListMultiline(self):
        self.oneTest("""
            from X import A, B, \\
                 C, D, \\
                 E
            """, """
            from X import A
            from X import B
            from X import C
            from X import D
            from X import E
            """)