def test_load_mask_sites_recognizes_non_bed_file(self, m_read_mask_file):
"""load_mask_sites should pass any other files to read_mask_file"""
m_read_mask_file.return_value = [5, 6]
assert utils.load_mask_sites("mask.not_a_bed")
m_read_mask_file.assert_called_with("mask.not_a_bed")
def test_load_mask_sites_recognizes_bed_file(self, m_read_bed_file,
extension):
"""load_mask_sites should handle files that end with .bed with any capitalization as a bed"""
m_read_bed_file.return_value = [3, 4]
assert utils.load_mask_sites("mask.%s" % extension) == [3, 4]
m_read_bed_file.assert_called_with("mask.%s" % extension)
