def test_get_line_ending(self): self.assertEqual('\n', autoflake.get_line_ending('\n')) self.assertEqual('\n', autoflake.get_line_ending('abc\n')) self.assertEqual('\t \t\n', autoflake.get_line_ending('abc\t \t\n')) self.assertEqual('', autoflake.get_line_ending('abc')) self.assertEqual('', autoflake.get_line_ending(''))