def test_filter_unused_variable_with_basic_data_structures(self): self.assertEqual('pass', autoflake.filter_unused_variable('x = dict()')) self.assertEqual('pass', autoflake.filter_unused_variable('x = list()')) self.assertEqual('pass', autoflake.filter_unused_variable('x = set()'))
def test_filter_unused_variable_with_exception(self): self.assertEqual( 'except Exception:', autoflake.filter_unused_variable('except Exception as exception:')) self.assertEqual( 'except (ImportError, ValueError):', autoflake.filter_unused_variable( 'except (ImportError, ValueError) as foo:'))
def test_filter_unused_variable_with_literal_or_name(self): self.assertEqual('pass', autoflake.filter_unused_variable('x = 1')) self.assertEqual('pass', autoflake.filter_unused_variable('x = y')) self.assertEqual('pass', autoflake.filter_unused_variable('x = {}'))
def test_filter_unused_variable(self): self.assertEqual('foo()', autoflake.filter_unused_variable('x = foo()')) self.assertEqual(' foo()', autoflake.filter_unused_variable(' x = foo()'))
def test_filter_unused_variable_should_multiple_assignments(self): self.assertEqual('x = y = foo()', autoflake.filter_unused_variable('x = y = foo()'))
def test_filter_unused_variable_should_ignore_multiline(self): self.assertEqual('x = foo()\\', autoflake.filter_unused_variable('x = foo()\\'))