def test_expand_raises_with_missing_contig_in_map(self):
# Empty contig_map should raise.
with self.assertRaises(KeyError):
ranges.expand(ranges.make_range('1', 10, 20), 1, contig_map={})
# Missing '1' from the contig map should raise.
with self.assertRaises(KeyError):
ranges.expand(ranges.make_range('1', 10, 20),
1,
contig_map={
'2': reference_pb2.ContigInfo(name='2',
n_bases=50),
})
def test_expand_handles_boundaries(self, region, n_bp, contig_map,
expected):
self.assertEqual(expected, ranges.expand(region, n_bp, contig_map))
def test_expand_raises_on_negative_n_bp(self):
with self.assertRaisesRegexp(ValueError,
'n_bp must be >= 0 but got -10'):
ranges.expand(ranges.make_range('1', 10, 20), -10)
def test_expand_is_correct(self, region, n_bp, contig_map, expected):
self.assertEqual(expected, ranges.expand(region, n_bp, contig_map))
