diff --git a/pycaret/tests/test_preprocess.py b/pycaret/tests/test_preprocess.py new file mode 100644 index 000000000..78e9f1204 --- /dev/null +++ b/pycaret/tests/test_preprocess.py @@ -0,0 +1,23 @@ +import os, sys +sys.path.insert(0, os.path.abspath("..")) + +import pytest +import pycaret.datasets +import pycaret.preprocess + +def test(): + # loading dataset + data = pycaret.datasets.get_data("juice") + target = "Purchase" + + # preprocess all in one + X = pycaret.preprocess.Preprocess_Path_One(train_data=data, target_variable=target, display_types=False) + + # preprocess all in one unsupervised + X = pycaret.preprocess.Preprocess_Path_Two(train_data=data, display_types=False) + + assert 1 == 1 + +if __name__ == "__main__": + test() +