class LambTests(unittest.TestCase):

    def test_lamb_output(self):
        outfile = StringIO()
lamb.write_lamb(outfile)
outfile.seek(0)
content = outfile.read()
self.assertEqual(content, 'Mary had a little lamb.\n')
Как выполнить модульное тестирование функций записи файлов с помощью python unittest