def test_adds_final_newline_if_necessary(self): s = Source() s.update('new stuff') self.assertEqual(s.get_updated_contents(), 'new stuff\n')
def test_update_with_empty_contents(self): s = Source() s.update('new stuff\n') self.assertEqual(s.get_updated_contents(), 'new stuff\n')