Fix check for TYPE_CHECKING (#207)

* Fix check for TYPE_CHECKING

This will unbreak default branch builds

* Un-autoformat test case files

Also remove StringIO

Author: sigmavirus24

flake8_import_order/__init__.py

@@ -142,7 +142,15 @@ def visit_ImportFrom(self, node):  # noqa: N802
     def _type_checking_import(self, node):
         return (
             isinstance(node.parent, ast.If)
-            and node.parent.test.id == "TYPE_CHECKING"
+            and isinstance(node.parent.test, ast.Name)
+            and (
+                node.parent.test.id == "TYPE_CHECKING"
+                or (
+                    node.parent.test.value.id in {"t", "typing"}
+                    and getattr(node.parent.test, "attr", "")
+                    == "TYPE_CHECKING"
+                )
+            )
         )
 
     def _classify_type(self, module):

tests/test_cases/complete_appnexus.py

@@ -38,7 +38,6 @@
     from functools import *
     import os
     from os import path
-    import StringIO
     import sys
 
     import X
@@ -67,3 +66,23 @@
     from .. import B
     from ..A import A
     from ..B import B
+
+if t.TYPE_CHECKING:
+    import ast
+
+    import X
+
+    from localpackage import A, b
+
+    import flake8_import_order
+    from ..B import B
+
+if typing.TYPE_CHECKING:
+    import ast
+
+    import X
+
+    from localpackage import A, b
+
+    import flake8_import_order
+    from ..B import B

tests/test_cases/complete_cryptography.py

@@ -82,3 +82,23 @@
     from .. import B
     from ..A import A
     from ..B import B
+
+if t.TYPE_CHECKING:
+    import ast
+
+    import X
+
+    from localpackage import A, b
+
+    import flake8_import_order
+    from ..B import B
+
+if typing.TYPE_CHECKING:
+    import ast
+
+    import X
+
+    from localpackage import A, b
+
+    import flake8_import_order
+    from ..B import B

tests/test_cases/complete_edited.py

@@ -39,7 +39,6 @@
 if TYPE_CHECKING:
     import ast
     import os
-    import StringIO
     import sys
     from functools import *
     from os import path
@@ -72,4 +71,55 @@
     from .. import A
     from .. import B
     from ..A import A
-    from ..B import B
+    from ..B import B
+
+if t.TYPE_CHECKING:
+    import ast
+
+    import X
+
+    from localpackage import A, b
+
+    import flake8_import_order
+    from ..B import B
+
+if typing.TYPE_CHECKING:
+    import ast
+    import os
+    import sys
+    from functools import *
+    from os import path
+
+    import localpackage
+    import X
+    import Y
+    import Z
+    from localpackage import A
+    from localpackage import b
+    from X import A
+    from X import B
+    from X import C
+    from X import *
+    from X import b
+    from X import d
+    from Y import A
+    from Y import B
+    from Y import C
+    from Y import D
+    from Y import *
+    from Y import e
+    from Z import A
+    from Z.A import A
+    from Z.A.B import A
+
+    import flake8_import_order
+    from flake8_import_order import *
+
+    from .. import A
+    from .. import B
+    from ..A import A
+    from ..B import B
+    from . import A
+    from . import B
+    from .A import A
+    from .B import B

tests/test_cases/complete_google.py

@@ -37,7 +37,6 @@
     from functools import *
     import os
     from os import path
-    import StringIO
     import sys
 
     import localpackage

tests/test_cases/complete_pycharm.py

@@ -33,7 +33,6 @@
 from ..B import B
 
 if TYPE_CHECKING:
-    import StringIO
     import ast
     import os
     import sys

tests/test_cases/complete_smarkets.py

@@ -35,7 +35,6 @@
 if TYPE_CHECKING:
     import ast
     import os
-    import StringIO
     import sys
     from functools import *
     from os import path

tests/test_pylama_linter.py

@@ -1,7 +1,10 @@
-from flake8_import_order.pylama_linter import Linter
+import pytest
 
 
+@pytest.mark.skip("pylama seems unmaintained")
 def test_linter(tmpdir):
+    from flake8_import_order.pylama_linter import Linter
+
     file_ = tmpdir.join("flake8_import_order.py")
     file_.write("import ast\nimport flake8_import_order\n")
 

tox.ini

@@ -1,10 +1,9 @@
 [tox]
-envlist = py3{7,8,9,10,11},pypy3,manifest,pep8,setuppy
+envlist = py3{9,10,11,12,13},pypy3,pep8,release-check
 
 [testenv]
 deps =
     flake8
-    pylama
     pytest
     pytest-cov
 commands =