From fa11d8bbbc747b3e2985e6d077a552d0bb88c3ea Mon Sep 17 00:00:00 2001 From: Andrey Stebenkov Date: Mon, 19 Aug 2024 18:39:27 +0300 Subject: [PATCH] Update test with adding extra types_encountered --- test/unit/preprocessing/test_preprocessors.py | 17 +++++++++++++++-- 1 file changed, 15 insertions(+), 2 deletions(-) diff --git a/test/unit/preprocessing/test_preprocessors.py b/test/unit/preprocessing/test_preprocessors.py index 6eda74361d..d4d52c4884 100644 --- a/test/unit/preprocessing/test_preprocessors.py +++ b/test/unit/preprocessing/test_preprocessors.py @@ -216,8 +216,14 @@ def test_binary_pseudo_string_column_process_correctly(): pipeline = correct_preprocessing_params(pipeline) train_predicted = pipeline.fit(train_data) + types_encountered = ( + int, float, + np.int8, np.int16, np.int32, np.int64, + np.float16, np.float32, np.float64, + ) + assert train_predicted.features.shape[1] == 1 - assert all(isinstance(el[0], float) for el in train_predicted.features) + assert all(isinstance(el[0], types_encountered) for el in train_predicted.features) def fit_predict_cycle_for_testing(idx: int): @@ -240,8 +246,15 @@ def test_mixed_column_with_str_and_float_values(): # column with index 1 must be converted to float and the gaps must be filled train_predicted = fit_predict_cycle_for_testing(idx=1) + + types_encountered = ( + int, float, + np.int8, np.int16, np.int32, np.int64, + np.float16, np.float32, np.float64, + ) + assert train_predicted.features.shape[1] == 1 - assert all(isinstance(el[0], float) for el in train_predicted.features) + assert all(isinstance(el[0], types_encountered) for el in train_predicted.features) # column with index 2 must be removed due to unclear type of data try: