def test_container_gz(self):
"""Check the type gz of the container of an archive"""
mbox = MBoxArchive(self.cfiles['gz'])
container = mbox.container
self.assertIsInstance(container, gzip.GzipFile)
container.close()
def test_container_bz2(self):
"""Check the type bz2 of the container of an archive"""
mbox = MBoxArchive(self.cfiles['bz2'])
container = mbox.container
self.assertIsInstance(container, bz2.BZ2File)
container.close()
def test_container(self):
"""Check the type of the container of an archive"""
import _io
mbox = MBoxArchive(self.files['single'])
container = mbox.container
self.assertIsInstance(container, _io.BufferedReader)
container.close()
def test_compressed(self):
"""Test compressed properties"""
mbox = MBoxArchive(self.cfiles['bz2'])
self.assertEqual(mbox.compressed_type, 'bz2')
self.assertEqual(mbox.is_compressed(), True)
mbox = MBoxArchive(self.cfiles['gz'])
self.assertEqual(mbox.compressed_type, 'gz')
self.assertEqual(mbox.is_compressed(), True)
mbox = MBoxArchive(self.cfiles['zip'])
self.assertEqual(mbox.compressed_type, 'zip')
self.assertEqual(mbox.is_compressed(), True)
def test_container(self):
"""Check the type of the container of an archive"""
mbox = MBoxArchive(self.cfiles['bz2'])
container = mbox.container
self.assertIsInstance(container, bz2.BZ2File)
container.close()
mbox = MBoxArchive(self.cfiles['gz'])
container = mbox.container
self.assertIsInstance(container, gzip.GzipFile)
container.close()
import _io
mbox = MBoxArchive(self.files['single'])
container = mbox.container
self.assertIsInstance(container, _io.BufferedReader)
container.close()
def test_container_zip(self):
"""Check the type zip of the container of an archive"""
mbox = MBoxArchive(self.cfiles['zip'])
container = mbox.container
self.assertIsInstance(container, zipfile.ZipExtFile)
container.close()
with zipfile.ZipFile(self.cfiles['zip'], 'w') as f_out:
f_out.write(os.path.join(os.path.dirname(os.path.abspath(__file__)), 'data/mbox/mbox_single.mbox'))
f_out.write(os.path.join(os.path.dirname(os.path.abspath(__file__)), 'data/mbox/mbox_no_fields.mbox'))
mbox = MBoxArchive(self.cfiles['zip'])
with self.assertLogs(logger, level='ERROR') as cm:
container = mbox.container
container.close()
self.assertEqual(cm.output[0], 'ERROR:perceval.backends.core.mbox:Zip %s contains more than one file, '
'only the first uncompressed' % mbox.filepath)
def test_filepath(self):
"""Test filepath property"""
mbox = MBoxArchive(self.cfiles['gz'])
self.assertEqual(mbox.filepath, self.cfiles['gz'])
def test_not_compressed(self):
"""Check the properties of a non-compressed archive"""
mbox = MBoxArchive(self.files['single'])
self.assertEqual(mbox.compressed_type, None)
self.assertEqual(mbox.is_compressed(), False)
