Coverage for tests/utils/test_sklearn_wrapper.py: 100%

1from sklearn.svm import OneClassSVM 

2from sklearn.covariance import EllipticEnvelope 

3from sklearn.ensemble import IsolationForest 

4from sklearn.linear_model import SGDOneClassSVM 

5from sklearn.neighbors import LocalOutlierFactor 

6 

7from tadkit.base.tadlearner import TADLearner 

8from tadkit.utils.tadlearner_factory import tadlearner_factory 

9 

10 

11class TestClassicModels: 

12 @staticmethod 

13 def _test(Model): 

14 Learner = tadlearner_factory(Model, [], {}) 

15 Learner() 

16 # assert not isinstance(Model, TADLearner) 

17 assert isinstance(Learner, TADLearner) 

18 

19 def test_1(self): 

20 self._test(OneClassSVM) 

21 

22 def test_2(self): 

23 self._test(SGDOneClassSVM) 

24 

25 def test_3(self): 

26 self._test(EllipticEnvelope) 

27 

28 def test_4(self): 

29 self._test(LocalOutlierFactor) 

30 

31 def test_5(self): 

32 self._test(IsolationForest)