Handle UnicodeDecodeError

autoflake.py

@@ -84,17 +84,11 @@ def unused_import_line_numbers(source):
 def check(source):
     """Return messages from pyflakes."""
     reporter = ListReporter()
-    pyflakes.api.check(source, filename='<string>', reporter=reporter)
-    return reporter.messages
-
-
-class StubFile(object):
-
-    """Stub for ignoring output."""
-
-    def write(_, __):
-        """Stub write()."""
+    try:
+        pyflakes.api.check(source, filename='<string>', reporter=reporter)
+    except UnicodeDecodeError:
         pass
+    return reporter.messages
 
 
 class ListReporter(pyflakes.reporter.Reporter):
@@ -107,7 +101,7 @@ def __init__(self):
         Ignore errors from Reporter.
 
         """
-        ignore = StubFile()
+        ignore = io.StringIO()
         pyflakes.reporter.Reporter.__init__(self, ignore, ignore)
         self.messages = []
 

test_autoflake.py

@@ -1,4 +1,5 @@
 #!/usr/bin/env python
+# coding: utf-8
 
 """Test suite for autoflake."""
 
@@ -439,6 +440,9 @@ def test_check(self):
     def test_check_with_bad_syntax(self):
         self.assertFalse(autoflake.check('foo('))
 
+    def test_check_with_unicode(self):
+        self.assertFalse(autoflake.check('print("∑"'))
+
 
 class SystemTests(unittest.TestCase):