def test_erlich_zielinski_dnarules():
dir_path = os.getcwd()
try:
os.remove(dir_path + "/DEC_" + out_dir2)
except:
print("Not deleting, File did not exists")
shutil.copyfile(dir_path + "/" + cmp_file2, dir_path + "/" + file2)
chunksize = 75
number_of_chunks = Encoder.get_number_of_chunks_for_file_with_chunk_size(
file2, chunksize)
pseudo_decoder = LTDecoder.pseudo_decoder(number_of_chunks)
dist = ErlichZielinskiRobustSolitonDistribution(k=number_of_chunks,
delta=0.2,
seed=2)
rules = DNARules_ErlichZielinski()
encoder = LTEncoder(file2,
number_of_chunks,
dist,
error_correction=reed_solomon_encode,
chunk_size=chunksize,
rules=rules,
insert_header=False,
number_of_chunks_len_format="H",
id_len_format="H",
used_packets_len_format="H",
pseudo_decoder=pseudo_decoder,
implicit_mode=True,
save_number_of_chunks_in_packet=False)
encoder.encode_to_packets()
encoder.save_packets(split_to_multiple_files=True, save_as_dna=True)
assert pseudo_decoder.is_decoded() and pseudo_decoder.getSolvedCount(
) == encoder.number_of_chunks
assert os.path.exists(out_dir2)
decoder = LTDecoder(out_dir2,
use_headerchunk=False,
dist=dist,
implicit_mode=True,
static_number_of_chunks=encoder.number_of_chunks,
error_correction=reed_solomon_decode)
decoder.decodeFolder(number_of_chunks_len_format="H",
seed_len_format="H",
degree_len_format="H")
assert decoder.is_decoded() and decoder.getSolvedCount(
) == encoder.number_of_chunks
decoder.saveDecodedFile(print_to_output=False, null_is_terminator=True)
out_file2 = "DEC_LT_" + file2
assert os.path.exists(out_file2) and filecmp.cmp(out_file2, cmp_file2)
os.remove(out_file2)
def test_suite(as_dna, decoder_instance, distribution, use_header,
implicit_mode):
dir_path = os.getcwd()
try:
os.remove(dir_path + "/" + file)
except:
print("Not deleting, File did not exists")
shutil.copyfile(dir_path + "/" + cmp_file, dir_path + "/" + file)
chunksize = 200
number_of_chunks = Encoder.get_number_of_chunks_for_file_with_chunk_size(
file, chunksize)
pseudo_decoder = decoder_instance.pseudo_decoder(number_of_chunks)
if distribution == "robust":
dist = RobustSolitonDistribution(S=number_of_chunks, delta=0.2, seed=2)
elif distribution == "ideal":
dist = IdealSolitonDistribution(S=number_of_chunks, seed=2)
else:
dist = ErlichZielinskiRobustSolitonDistribution(k=number_of_chunks,
delta=0.2,
seed=2)
rules = FastDNARules() if as_dna else None
encoder = LTEncoder(file,
number_of_chunks,
dist,
chunk_size=chunksize,
pseudo_decoder=pseudo_decoder,
rules=rules,
insert_header=use_header,
number_of_chunks_len_format="H",
id_len_format="H",
used_packets_len_format="H",
implicit_mode=implicit_mode)
encoder.encode_to_packets()
encoder.save_packets(split_to_multiple_files=True, save_as_dna=as_dna)
assert pseudo_decoder.is_decoded() and pseudo_decoder.getSolvedCount(
) == encoder.number_of_chunks
assert os.path.exists(out_dir)
decoder = decoder_instance(out_dir,
use_headerchunk=use_header,
dist=dist,
implicit_mode=implicit_mode)
decoder.decodeFolder(number_of_chunks_len_format="H",
seed_len_format="H",
degree_len_format="H")
assert decoder.is_decoded() and decoder.getSolvedCount(
) == encoder.number_of_chunks
os.remove(file)
decoder.saveDecodedFile(print_to_output=False)
if not use_header:
out_file = "DEC_LT_" + file
else:
out_file = file
assert os.path.exists(out_file) and filecmp.cmp(out_file, cmp_file)
if decoder_instance == LTBPDecoder:
# since ApproxDecoder defines an upper bound Gauss-Decoder MUST be able to decode!
decoder = LTDecoder(out_dir,
use_headerchunk=use_header,
dist=dist,
implicit_mode=implicit_mode)
decoder.decodeFolder(number_of_chunks_len_format="H",
seed_len_format="H",
degree_len_format="H")
assert (decoder.is_decoded()
and decoder.getSolvedCount() == encoder.number_of_chunks)
os.remove(out_file)
decoder.saveDecodedFile(print_to_output=False)
assert os.path.exists(out_file) and filecmp.cmp(out_file, cmp_file)
shutil.rmtree(out_dir)