def test_binarize_labels_multiclass(self):
md = pd.Series([c for c in 'abcabc'])
labels = _binarize_labels(md, ['a', 'b', 'c'])
exp = np.array([[1, 0, 0], [0, 1, 0], [0, 0, 1],
[1, 0, 0], [0, 1, 0], [0, 0, 1]])
np.testing.assert_array_equal(exp, labels)
def test_binarize_labels_binary(self):
md = pd.Series([c for c in 'aabbaa'])
labels = _binarize_labels(md, ['a', 'b'])
exp = np.array([[1, 0], [1, 0], [0, 1], [0, 1], [1, 0], [1, 0]])
np.testing.assert_array_equal(exp, labels)