Can you also add tests that exercise the cases where an invalid column name is specified, as well as when the number of classes is the same as the number of sample IDs?