Please add a test for the multi-output case (that is when Y is a 2D boolean indicator matrix).