def test_append(): with create_tmp_files() as (tdir, in_file, out_file, dcmp_file): create_array(1, in_file) pack_file(in_file, out_file) append(out_file, in_file) unpack_file(out_file, dcmp_file) in_content = open(in_file, 'rb').read() dcmp_content = open(dcmp_file, 'rb').read() nt.assert_equal(len(dcmp_content), len(in_content) * 2) nt.assert_equal(dcmp_content, in_content * 2)
def pack_unpack(repeats, chunk_size=None, progress=False): with create_tmp_files() as (tdir, in_file, out_file, dcmp_file): if progress: print("Creating test array") create_array(repeats, in_file, progress=progress) if progress: print("Compressing") pack_file(in_file, out_file, chunk_size=chunk_size) if progress: print("Decompressing") unpack_file(out_file, dcmp_file) if progress: print("Verifying") cmp_file(in_file, dcmp_file)