By setting, the `labels=range(3)`, you assume that all test will have labels `[0, 1, 2]`.