def test_filter_code(self): self.assertEqual( """\ import os pass os.foo() """, ''.join(autoflake.filter_code("""\ import os import re os.foo() """)))
def test_filter_code(self): self.assertEqual( """\ import os pass os.foo() """, ''.join(autoflake.AutoFlake().filter_code("""\ import os import re os.foo() """)))
def test_filter_code_should_ignore_semicolons(self): self.assertEqual( r"""\ import os pass import os; import math, subprocess os.foo() """, ''.join( autoflake.filter_code(r"""\ import os import re import os; import math, subprocess os.foo() """)))
def test_filter_code_should_ignore_semicolons(self): self.assertEqual( r"""\ import os pass import os; import math, subprocess os.foo() """, ''.join(autoflake.AutoFlake().filter_code(r"""\ import os import re import os; import math, subprocess os.foo() """)))
def test_filter_code_with_indented_import(self): self.assertEqual( """\ import os if True: pass os.foo() """, ''.join(autoflake.AutoFlake().filter_code("""\ import os if True: import re os.foo() """)))
def test_filter_code_with_indented_import(self): self.assertEqual( """\ import os if True: pass os.foo() """, ''.join( autoflake.filter_code("""\ import os if True: import re os.foo() """)))
def test_filter_code_should_ignore_multiline_imports(self): self.assertEqual( r"""\ import os pass import os, \ math, subprocess os.foo() """, ''.join( autoflake.filter_code(r"""\ import os import re import os, \ math, subprocess os.foo() """)))
def test_filter_code_should_ignore_multiline_imports(self): self.assertEqual( r"""\ import os pass import os, \ math, subprocess os.foo() """, ''.join(autoflake.filter_code(r"""\ import os import re import os, \ math, subprocess os.foo() """)))
def test_filter_useless_pass_with_try(self): self.assertEqual( """\ import os os.foo() try: pass except ImportError: pass """, ''.join( autoflake.filter_useless_pass("""\ import os os.foo() try: pass pass except ImportError: pass """)))
def test_fix_code(self): self.assertEqual( """\ import os import math from sys import version os.foo() math.pi x = version """, autoflake.fix_code("""\ import os import re import abc, math, subprocess from sys import exit, version os.foo() math.pi x = version """))
def test_fix_code(self): self.assertEqual( """\ import os import math from sys import version os.foo() math.pi x = version """, autoflake.AutoFlake().fix_code("""\ import os import re import abc, math, subprocess from sys import exit, version os.foo() math.pi x = version """))
def test_filter_code_should_ignore_non_standard_library(self): self.assertEqual( """\ import os import my_own_module pass from my_package import another_module from my_package import subprocess from my_blah.my_blah_blah import blah os.foo() """, ''.join( autoflake.filter_code("""\ import os import my_own_module import re from my_package import another_module from my_package import subprocess from my_blah.my_blah_blah import blah os.foo() """)))
def test_filter_useless_pass_with_try(self): self.assertEqual( """\ import os os.foo() try: pass except ImportError: pass """, ''.join(autoflake.AutoFlake().filter_useless_pass( """\ import os os.foo() try: pass pass except ImportError: pass """)))
def test_filter_code_should_ignore_non_standard_library(self): self.assertEqual( """\ import os import my_own_module pass from my_package import another_module from my_package import subprocess from my_blah.my_blah_blah import blah os.foo() """, ''.join(autoflake.AutoFlake().filter_code("""\ import os import my_own_module import re from my_package import another_module from my_package import subprocess from my_blah.my_blah_blah import blah os.foo() """)))