def test_single_bz2_file(self):
filename = u"tests/testdata/simple_1.concrete.bz2"
reader = CommunicationReader(filename)
(comm1, comm1_filename) = reader.next()
self.assertTrue(hasattr(comm1, "sentenceForUUID"))
self.assertEqual(u"one", comm1.id)
self.assertEqual(filename, comm1_filename)
def test_single_gz_file_no_add_references(self):
filename = u"tests/testdata/simple_1.concrete.gz"
reader = CommunicationReader(filename, add_references=False)
(comm1, comm1_filename) = reader.next()
self.assertFalse(hasattr(comm1, "sentenceForUUID"))
self.assertEqual(u"one", comm1.id)
self.assertEqual(filename, comm1_filename)
def test_explicit_single_gz_file(self):
filename = u"tests/testdata/simple_1.concrete.gz"
reader = CommunicationReader(filename, filetype=FileType.STREAM_GZ)
(comm1, comm1_filename) = reader.next()
self.assertTrue(hasattr(comm1, "sentenceForUUID"))
self.assertEqual(u"one", comm1.id)
self.assertEqual(filename, comm1_filename)
def test_single_bz2_file(self):
filename = u'tests/testdata/simple_1.concrete.bz2'
reader = CommunicationReader(filename)
(comm1, comm1_filename) = reader.next()
self.assertTrue(hasattr(comm1, 'sentenceForUUID'))
self.assertEqual(u'one', comm1.id)
self.assertEqual(filename, comm1_filename)
def test_single_gz_file_no_add_references(self):
filename = u'tests/testdata/simple_1.concrete.gz'
reader = CommunicationReader(filename, add_references=False)
(comm1, comm1_filename) = reader.next()
self.assertFalse(hasattr(comm1, 'sentenceForUUID'))
self.assertEqual(u'one', comm1.id)
self.assertEqual(filename, comm1_filename)
def test_explicit_single_gz_file(self):
filename = u'tests/testdata/simple_1.concrete.gz'
reader = CommunicationReader(filename, filetype=FileType.STREAM_GZ)
(comm1, comm1_filename) = reader.next()
self.assertTrue(hasattr(comm1, 'sentenceForUUID'))
self.assertEqual(u'one', comm1.id)
self.assertEqual(filename, comm1_filename)
