def test_pandas_confusion_cm_binarize():
y_true = ['rabbit', 'cat', 'rabbit', 'rabbit', 'cat', 'dog', 'dog', 'rabbit', 'rabbit', 'cat', 'dog', 'rabbit']
y_pred = ['cat', 'cat', 'rabbit', 'dog', 'cat', 'rabbit', 'dog', 'cat', 'rabbit', 'cat', 'rabbit', 'rabbit']
cm = ConfusionMatrix(y_true, y_pred)
print("Confusion matrix:\n%s" % cm)
select = ['cat', 'dog']
print("Binarize with %s" % select)
binary_cm = cm.binarize(select)
print("Binary confusion matrix:\n%s" % binary_cm)
assert cm.sum() == binary_cm.sum()
