def test_write_footer(self): """ The write_footer method produces the correct string. """ writer = KiCAD() buf = StringIO() writer.write_footer(buf) self.assertEqual(buf.getvalue(), '$EndSCHEMATC\n')