def __test__(self, script, string): file_ = StringIO() pickler = ScriptPickler(file_) pickler.dump(script) self.assertEqual(file_.getvalue(), string) file_ = StringIO() pickler = ScriptPickler() pickler.dump(script, file=file_) self.assertEqual(file_.getvalue(), string) self.assertEqual(pickler.dumps(script), string)