Skip to content

Commit

Permalink
Create test_preprocess.py (#409)
Browse files Browse the repository at this point in the history
  • Loading branch information
daikikatsuragawa committed Aug 10, 2020
1 parent 268efe6 commit 6423d76
Showing 1 changed file with 23 additions and 0 deletions.
23 changes: 23 additions & 0 deletions 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()

0 comments on commit 6423d76

Please sign in to comment.