def test_doc_save_and_load(self): filepath = os.path.join(self.tempdir, 'test_doc_save_and_load.pkl') self.doc.save(filepath) new_doc = Doc.load(filepath) self.assertIsInstance(new_doc, Doc) self.assertEqual(len(new_doc), len(self.doc)) self.assertEqual(new_doc.lang, self.doc.lang) self.assertEqual(new_doc.metadata, self.doc.metadata)
def test_doc_save_and_load(tmpdir, doc): filepath = str(tmpdir.join('test_doc_save_and_load.pkl')) doc.save(filepath) new_doc = Doc.load(filepath) assert isinstance(new_doc, Doc) assert len(new_doc) == len(doc) assert new_doc.lang == doc.lang assert new_doc.metadata == doc.metadata