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')
