Added unittest for _get_bases_mask