示例#1
0
    def _fix_flake_import_errors(self) -> None:
        """Fix python source code to correct missed or unused import statements."""
        error_messages = autoflake.check(self._join_code())

        for message in error_messages:
            if isinstance(message, (UndefinedName, UndefinedExport)):
                object_name = message.message_args[0]
                self._add_package(object_name)
            elif isinstance(message, UnusedImport):
                import_name = message.message_args[0]
                self._remove_unused_imports(import_name)
示例#2
0
def pyflakes_count(filename):
    """Return pyflakes error count."""
    with autoflake.open_with_encoding(
            filename, encoding=autoflake.detect_encoding(filename)) as f:
        return len(list(autoflake.check(f.read())))
示例#3
0
    def test_check_with_unicode(self):
        self.assertFalse(autoflake.check('print("∑")'))

        self.assertTrue(autoflake.check('import os  # ∑'))
示例#4
0
 def test_check_with_bad_syntax(self):
     self.assertFalse(autoflake.check('foo('))
示例#5
0
 def test_check(self):
     self.assertTrue(autoflake.check('import os'))
示例#6
0
 def test_unused_import_line_numbers_with_dot(self):
     self.assertEqual([1],
                      list(
                          autoflake.unused_import_line_numbers(
                              autoflake.check('import os.path\n'))))
示例#7
0
 def test_unused_import_line_numbers_with_from(self):
     self.assertEqual([1],
                      list(
                          autoflake.unused_import_line_numbers(
                              autoflake.check('from os import path\n'))))
示例#8
0
    def test_check_with_unicode(self):
        self.assertFalse(autoflake.check('print("∑")'))

        self.assertTrue(autoflake.check('import os  # ∑'))
示例#9
0
 def test_check_with_bad_syntax(self):
     self.assertFalse(autoflake.check('foo('))
示例#10
0
 def test_check(self):
     self.assertTrue(autoflake.check('import os'))
示例#11
0
 def test_unused_import_line_numbers_with_dot(self):
     self.assertEqual(
         [1],
         list(autoflake.unused_import_line_numbers(
             autoflake.check('import os.path\n'))))
示例#12
0
 def test_unused_import_line_numbers_with_from(self):
     self.assertEqual(
         [1],
         list(autoflake.unused_import_line_numbers(
             autoflake.check('from os import path\n'))))
示例#13
0
def pyflakes_count(filename):
    """Return pyflakes error count."""
    with autoflake.open_with_encoding(
            filename,
            encoding=autoflake.detect_encoding(filename)) as f:
        return len(list(autoflake.check(f.read())))
示例#14
0
 def test_check_with_unicode(self):
     self.assertFalse(autoflake.check('print("∑")'))