Data science/mabdeljalil V1

This isn't actually enforced in the code. I could easily pass "C" and it would work. This is basically tautological since it only verifies the thing you just passed in.

If you added validation that the cohort were either A/B (which I don't particularly think you should), then you'd have a negative test to make sure that the class raises in that case.