|
- import numpy as np
- import pandas as pd
- import os
- from tods.sk_interface.feature_analysis.StatisticalAbsSum_skinterface import StatisticalAbsSumSKI
-
- from pyod.utils.data import generate_data
- import unittest
- from sklearn.utils.testing import assert_allclose
- from sklearn.utils.testing import assert_array_less
- from sklearn.utils.testing import assert_equal
- from sklearn.utils.testing import assert_greater
- from sklearn.utils.testing import assert_greater_equal
- from sklearn.utils.testing import assert_less_equal
- from sklearn.utils.testing import assert_raises
- from sklearn.metrics import roc_auc_score
-
- class StatisticalAbsSumSKI_TestCase(unittest.TestCase):
- def setUp(self):
-
- self.n_train = 200
- self.n_test = 100
- self.X_train, self.y_train, self.X_test, self.y_test = generate_data(
- n_train=self.n_train, n_test=self.n_test, n_features=5,
- contamination=0., random_state=42)
-
- self.transformer = StatisticalAbsSumSKI()
-
- def test_produce(self):
- X_transform = self.transformer.produce(self.X_test)
-
-
-
- if __name__ == '__main__':
- unittest.main()
|