def test_write_qual_report(self):
""" Writes data to output text file properly """
output_dir = self.output_dir
qual_bins = [[1, 2, 6], [1, 2, 3], [2, 4], [4]]
expected_ave_bins = [3.00, 2.00, 3.00, 4.00]
expected_std_dev_bins = [2.16, 0.816, 1.0, 0]
expected_total_bases_bins = [3, 3, 2, 1]
score_min = 25
write_qual_report(expected_ave_bins, expected_std_dev_bins,
expected_total_bases_bins, output_dir, score_min)
# Test text file output for proper data
text_output_fp = output_dir + "quality_bins.txt"
text_output_f = open(text_output_fp, "U")
actual_text_output = [line.strip() for line in text_output_f]
ave_bin_index = 2
std_dev_bin_index = 4
total_bases_index = 6
actual_bins_ave =\
[float(f) for f in actual_text_output[ave_bin_index].split(',')]
actual_bins_std =\
[float(f) for f in actual_text_output[std_dev_bin_index].split(',')]
actual_bins_total_bases =\
[float(f) for f in actual_text_output[total_bases_index].split(',')]
self.assertEqual(actual_bins_ave, expected_ave_bins)
self.assertEqual(actual_bins_std, expected_std_dev_bins)
self.assertEqual(actual_bins_total_bases, expected_total_bases_bins)
def test_write_qual_report(self):
""" Writes data to output text file properly """
output_dir = self.output_dir
qual_bins = [[1, 2, 6], [1, 2, 3], [2, 4], [4]]
expected_ave_bins = [3.00, 2.00, 3.00, 4.00]
expected_std_dev_bins = [2.16, 0.816, 1.0, 0]
expected_total_bases_bins = [3, 3, 2, 1]
score_min = 25
write_qual_report(expected_ave_bins, expected_std_dev_bins,
expected_total_bases_bins, output_dir, score_min)
# Test text file output for proper data
text_output_fp = output_dir + "quality_bins.txt"
text_output_f = open(text_output_fp, "U")
actual_text_output = [line.strip() for line in text_output_f]
ave_bin_index = 2
std_dev_bin_index = 4
total_bases_index = 6
actual_bins_ave =\
[float(f) for f in actual_text_output[ave_bin_index].split(',')]
actual_bins_std =\
[float(f)
for f in actual_text_output[std_dev_bin_index].split(',')]
actual_bins_total_bases =\
[float(f)
for f in actual_text_output[total_bases_index].split(',')]
self.assertEqual(actual_bins_ave, expected_ave_bins)
self.assertEqual(actual_bins_std, expected_std_dev_bins)
self.assertEqual(actual_bins_total_bases, expected_total_bases_bins)
