class DocumentTest(unittest.TestCase): file_name = u'models/test_data/lorem.json' meta = {'title': u'test 稢綌', 'author': u'gorden 胇赲'} body = u'In id tristique orci. 痵痽 犵艿邔 疿疶砳 齸圞趲.' pre_file_name = file_name + '_PRE.json' raw_file_name = file_name def setUp(self): self.doc = Document(file_name=self.file_name, metadata=self.meta, pre_file_name=self.pre_file_name, raw_file_name=self.raw_file_name) def test_clone(self): """ Test cloning a document """ doc_cloned = self.doc.clone() self.assertEqual(doc_cloned.file_name, self.doc.file_name) self.assertEqual(doc_cloned.pre_file_name, self.doc.pre_file_name) self.assertEqual(doc_cloned.raw_file_name, self.doc.raw_file_name) self.assertEqual(doc_cloned.metadata, self.doc.metadata) self.assertEqual(doc_cloned.raw_body, self.doc.raw_body) self.assertEqual(self.doc, doc_cloned) # using assertFalse instead of assertNotEqual in order to # test __eq__ doc_cloned.file_name = u'nope' self.assertFalse(self.doc == doc_cloned) doc_cloned.file_name = self.doc.file_name doc_cloned.metadata = None self.assertFalse(self.doc == doc_cloned) doc_cloned.metadata = self.doc.metadata doc_cloned.raw_file_name = '' self.assertFalse(self.doc == doc_cloned) doc_cloned.raw_file_name = self.doc.raw_file_name doc_cloned.pre_file_name = '' self.assertFalse(self.doc == doc_cloned) def test_to_dict(self): """ Test conversion to dictionary (for json serialization) """ doc_dict = self.doc.to_dict() self.assertEqual(doc_dict['file_name'], self.doc.file_name) self.assertEqual(doc_dict['metadata'], self.doc.metadata) self.assertEqual(doc_dict['pre_file_name'], self.doc.pre_file_name) # TODO check raw def test_open(self): """ Test opening a Document json """ self.assertRaises(InvalidDocumentException, Document.from_json, 'models/test_data/invalid.json') self.assertRaises(InvalidDocumentException, Document.from_json, 'models/test_data/invalid.txt')
class DocumentTest(unittest.TestCase): file_name = u'models/test_data/lorem.json' meta = {'title': u'test 稢綌', 'author': u'gorden 胇赲' } body = u'In id tristique orci. 痵痽 犵艿邔 疿疶砳 齸圞趲.' pre_file_name = file_name + '_PRE.json' raw_file_name = file_name def setUp(self): self.doc = Document(file_name=self.file_name, metadata=self.meta, pre_file_name=self.pre_file_name, raw_file_name=self.raw_file_name) def test_clone(self): """ Test cloning a document """ doc_cloned = self.doc.clone() self.assertEqual(doc_cloned.file_name, self.doc.file_name) self.assertEqual(doc_cloned.pre_file_name, self.doc.pre_file_name) self.assertEqual(doc_cloned.raw_file_name, self.doc.raw_file_name) self.assertEqual(doc_cloned.metadata, self.doc.metadata) self.assertEqual(doc_cloned.raw_body, self.doc.raw_body) self.assertEqual(self.doc, doc_cloned) # using assertFalse instead of assertNotEqual in order to # test __eq__ doc_cloned.file_name = u'nope' self.assertFalse(self.doc == doc_cloned) doc_cloned.file_name = self.doc.file_name doc_cloned.metadata = None self.assertFalse(self.doc == doc_cloned) doc_cloned.metadata = self.doc.metadata doc_cloned.raw_file_name = '' self.assertFalse(self.doc == doc_cloned) doc_cloned.raw_file_name = self.doc.raw_file_name doc_cloned.pre_file_name = '' self.assertFalse(self.doc == doc_cloned) def test_to_dict(self): """ Test conversion to dictionary (for json serialization) """ doc_dict = self.doc.to_dict() self.assertEqual(doc_dict['file_name'], self.doc.file_name) self.assertEqual(doc_dict['metadata'], self.doc.metadata) self.assertEqual(doc_dict['pre_file_name'], self.doc.pre_file_name) # TODO check raw def test_open(self): """ Test opening a Document json """ self.assertRaises(InvalidDocumentException, Document.from_json, 'models/test_data/invalid.json') self.assertRaises(InvalidDocumentException, Document.from_json, 'models/test_data/invalid.txt')