Test LabelEncoder.fit_transform with arbitrary labels.

astaric · Aug 26, 2012 · 7107544 · 7107544
1 parent 56c6bfb
commit 7107544
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/sklearn/tests/test_preprocessing.py b/sklearn/tests/test_preprocessing.py
@@ -448,6 +448,10 @@ def test_label_encoder_fit_transform():
     ret = le.fit_transform([1, 1, 4, 5, -1, 0])
     assert_array_equal(ret, [2, 2, 3, 4, 0, 1])
 
+    le = LabelEncoder()
+    ret = le.fit_transform(["paris", "paris", "tokyo", "amsterdam"])
+    assert_array_equal(ret, [1, 1, 2, 0])
+
 
 def test_label_encoder_string_labels():
     """Test LabelEncoder's transform and inverse_transform methods with