def test_get_indentation(self): self.assertEqual('', autoflake.get_indentation('')) self.assertEqual(' ', autoflake.get_indentation(' abc')) self.assertEqual(' ', autoflake.get_indentation(' abc \n\t')) self.assertEqual('\t', autoflake.get_indentation('\tabc \n\t')) self.assertEqual(' \t ', autoflake.get_indentation(' \t abc \n\t')) self.assertEqual('', autoflake.get_indentation(' '))