Skip to content

Commit a94301c

Browse files
committed
added an additional test case in test_label.test_label_binarize_with_class_order()
1 parent 03dbec8 commit a94301c

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

sklearn/preprocessing/tests/test_label.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -410,6 +410,13 @@ def test_label_binarize_with_class_order():
410410
expected = np.array([[1, 0, 0, 0], [0, 1, 0, 0]])
411411
assert_array_equal(out, expected)
412412

413+
out = label_binarize([0, 1, 2, 3], classes=[3, 2, 0, 1])
414+
expected = np.array([[0, 0, 1, 0],
415+
[0, 0, 0, 1],
416+
[0, 1, 0, 0],
417+
[1, 0, 0, 0]])
418+
assert_array_equal(out, expected)
419+
413420

414421
def check_binarized_results(y, classes, pos_label, neg_label, expected):
415422
for sparse_output in [True, False]:

0 commit comments

Comments
 (0)