def test_preferences(self):
dobj = compressr.Opener()
self.assertEquals(["xz", "bz2", "gz"], dobj.preferences)
prefs = ["bz2", "gz"]
dobj = compressr.Opener(preferences=prefs)
self.assertEquals(prefs, dobj.preferences)
def _test_open(self, fname_uncompressed, with_uncompressed=True):
dobj = compressr.Opener()
file_names = [fname_uncompressed]
for ext in ['bzip2', 'xz', 'bz2', 'gz']:
fname = "foo.{}".format(ext)
file_names.append(fname)
# We don't want to explicitly test the writer at this point, but might
# as well
for fname in file_names:
if with_uncompressed:
kwargs = dict(uncompressed=(fname == fname_uncompressed))
else:
kwargs = {}
fobj = dobj.open(fname, "wb", **kwargs)
fobj.write(b"Test" * 100)
fobj.close()
fsize = os.stat(fname).st_size
if fname == fname_uncompressed:
self.assertEquals(400, fsize)
else:
self.assertTrue(fsize < 400)
for fname in file_names:
if with_uncompressed:
kwargs = dict(uncompressed=(fname == fname_uncompressed))
else:
kwargs = {}
fobj = dobj.open(fname, **kwargs)
self.assertEquals(b"Test" * 100, fobj.read())
def test_rank(self):
dobj = compressr.Opener()
fnames = [
"path1/Packages.gz", "path3/Packages.bz2", "path1/Packages.xz",
"path1/Packages.leave-me-alone", "path1/Packages", "path2/Packages"
]
ret = dobj.best_choice(fnames)
self.assertEquals(
['path1/Packages.xz', 'path2/Packages', 'path3/Packages.bz2'], ret)