diff --git a/src/python/nimbusml/tests/preprocessing/missing_values/test_data_with_missing.py b/src/python/nimbusml/tests/preprocessing/missing_values/test_data_with_missing.py index 99db58dd..012d7469 100644 --- a/src/python/nimbusml/tests/preprocessing/missing_values/test_data_with_missing.py +++ b/src/python/nimbusml/tests/preprocessing/missing_values/test_data_with_missing.py @@ -3,6 +3,7 @@ # Licensed under the MIT License. # -------------------------------------------------------------------------------------------- +import platform import unittest import numpy as np @@ -161,6 +162,7 @@ def test_input_conversion_to_float_retains_other_column_types(self): assert_equal(result.dtypes['f1'], np.object) assert_equal(result.dtypes['f2.f2'], np.float32) + @unittest.skipIf('centos' in platform.linux_distribution()[0].lower()) def test_category_imputation(self): data={'f0': [4, 4, np.nan, 9], 'f1': [4, 4, np.nan, np.nan]} diff --git a/src/python/nimbusml/tests/preprocessing/normalization/test_robustscaler.py b/src/python/nimbusml/tests/preprocessing/normalization/test_robustscaler.py index da854164..8511cc3b 100644 --- a/src/python/nimbusml/tests/preprocessing/normalization/test_robustscaler.py +++ b/src/python/nimbusml/tests/preprocessing/normalization/test_robustscaler.py @@ -2,6 +2,7 @@ # Copyright (c) Microsoft Corporation. All rights reserved. # Licensed under the MIT License. # -------------------------------------------------------------------------------------------- +import platform import unittest import pandas @@ -9,6 +10,7 @@ from nimbusml.preprocessing.normalization import RobustScaler +@unittest.skipIf('centos' in platform.linux_distribution()[0].lower()) class TestRobustScaler(unittest.TestCase): def test_with_integer_inputs(self): diff --git a/src/python/nimbusml/tests/preprocessing/test_datetimesplitter.py b/src/python/nimbusml/tests/preprocessing/test_datetimesplitter.py index aecfc20e..4407dd55 100644 --- a/src/python/nimbusml/tests/preprocessing/test_datetimesplitter.py +++ b/src/python/nimbusml/tests/preprocessing/test_datetimesplitter.py @@ -3,6 +3,7 @@ # Licensed under the MIT License. # -------------------------------------------------------------------------------------------- +import platform import unittest import pandas @@ -12,6 +13,7 @@ from sklearn.utils.testing import assert_equal +@unittest.skipIf('centos' in platform.linux_distribution()[0].lower()) class TestDateTimeSplitter(unittest.TestCase): def test_check_estimator_DateTimeSplitter(self): diff --git a/src/python/nimbusml/tests/preprocessing/test_tokeyimputer.py b/src/python/nimbusml/tests/preprocessing/test_tokeyimputer.py index 85c501c7..29661bf2 100644 --- a/src/python/nimbusml/tests/preprocessing/test_tokeyimputer.py +++ b/src/python/nimbusml/tests/preprocessing/test_tokeyimputer.py @@ -3,6 +3,7 @@ # Licensed under the MIT License. # -------------------------------------------------------------------------------------------- +import platform import unittest import numpy as np @@ -10,6 +11,7 @@ from nimbusml.preprocessing import ToKeyImputer +@unittest.skipIf('centos' in platform.linux_distribution()[0].lower()) class TestToKeyImputer(unittest.TestCase): def test_tokeyimputer(self): diff --git a/src/python/nimbusml/tests/preprocessing/test_tostring.py b/src/python/nimbusml/tests/preprocessing/test_tostring.py index edb11d63..e091a4a5 100644 --- a/src/python/nimbusml/tests/preprocessing/test_tostring.py +++ b/src/python/nimbusml/tests/preprocessing/test_tostring.py @@ -3,6 +3,7 @@ # Licensed under the MIT License. # -------------------------------------------------------------------------------------------- +import platform import unittest import numpy as np @@ -11,6 +12,7 @@ from sklearn.utils.testing import assert_equal +@unittest.skipIf('centos' in platform.linux_distribution()[0].lower()) class TestToString(unittest.TestCase): def test_tostring(self): diff --git a/src/python/nimbusml/tests/timeseries/test_timeseriesimputer.py b/src/python/nimbusml/tests/timeseries/test_timeseriesimputer.py index 98c9e21b..8a649f0e 100644 --- a/src/python/nimbusml/tests/timeseries/test_timeseriesimputer.py +++ b/src/python/nimbusml/tests/timeseries/test_timeseriesimputer.py @@ -3,6 +3,7 @@ # Licensed under the MIT License. # -------------------------------------------------------------------------------------------- +import platform import unittest import numpy as np @@ -10,6 +11,7 @@ from nimbusml.timeseries import TimeSeriesImputer +@unittest.skipIf('centos' in platform.linux_distribution()[0].lower()) class TestTimeSeriesImputer(unittest.TestCase): def test_timeseriesimputer_adds_new_row(self): diff --git a/src/python/tests/test_estimator_checks.py b/src/python/tests/test_estimator_checks.py index 4a4bd65f..18a5ba31 100644 --- a/src/python/tests/test_estimator_checks.py +++ b/src/python/tests/test_estimator_checks.py @@ -7,6 +7,7 @@ """ import json import os +import platform import unittest from nimbusml.cluster import KMeansPlusPlus @@ -282,6 +283,13 @@ 'TimeSeriesImputer' ]) +if 'centos' in platform.linux_distribution()[0].lower(): + skip_epoints |= set([ + 'DateTimeSplitter', + 'RobustScaler', + 'ToKeyImputer', + 'ToString']) + def load_json(file_path): with open(file_path) as f: