def fake_allele_counter(self, start_pos, counts):
allele_counter = mock.Mock()
allele_counter.summary_counts.return_value = [
deepvariant_pb2.AlleleCountSummary(ref_supporting_read_count=n_ref,
total_read_count=n_ref + n_alt,
ref_base=ref,
reference_name='chr1',
position=start_pos + i)
for i, (n_alt, n_ref, ref) in enumerate(counts)
]
return allele_counter
def fake_allele_counter(self, start_pos, counts):
allele_counter = mock.Mock()
# pylint: disable=g-complex-comprehension
allele_counter.summary_counts.return_value = [
deepvariant_pb2.AlleleCountSummary(ref_supporting_read_count=n_ref,
total_read_count=n_ref + n_alt,
ref_base=ref,
reference_name='chr1',
position=start_pos + i)
for i, (n_alt, n_ref, ref) in enumerate(counts)
]
allele_counter.counts.return_value = counts
# pylint: enable=g-complex-comprehension
return allele_counter
