Added two more tests for setting column subsets

@taldcroft @mhvk - does this add the two cases we need?