From b9cf96b8070d6fe3beefb85c231846b1d8808458 Mon Sep 17 00:00:00 2001 From: Oleksandr Pavlyk Date: Fri, 31 May 2019 08:23:08 -0500 Subject: [PATCH] MAINT: test_encoder_dtypes_pandas reads expected dtype from DF --- sklearn/preprocessing/tests/test_encoders.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sklearn/preprocessing/tests/test_encoders.py b/sklearn/preprocessing/tests/test_encoders.py index f14db9bc977a1..f7c8524378603 100644 --- a/sklearn/preprocessing/tests/test_encoders.py +++ b/sklearn/preprocessing/tests/test_encoders.py @@ -596,7 +596,7 @@ def test_encoder_dtypes_pandas(): assert_array_equal(enc.transform(X).toarray(), exp) X = pd.DataFrame({'A': [1, 2], 'B': ['a', 'b'], 'C': [3., 4.]}) - X_type = [int, object, float] + X_type = [X['A'].dtype, X['B'].dtype, X['C'].dtype] enc.fit(X) assert all([enc.categories_[i].dtype == X_type[i] for i in range(3)]) assert_array_equal(enc.transform(X).toarray(), exp)