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)