def test_pandas_confusion_cm_int(): y_true = [2, 0, 2, 2, 0, 1, 1, 2, 2, 0, 1, 2] y_pred = [0, 0, 2, 1, 0, 2, 1, 0, 2, 0, 2, 2] labels = ["ant", "bird", "cat"] cm = ConfusionMatrix(y_true, y_pred, labels=labels) print("Confusion matrix:\n%s" % cm) asserts(y_true, y_pred, cm) assert cm.len() == len(labels)