def test_rbf_tsvm_set_get_params(self): """ It checks that set_params and get_params works correctly for non-linear TSVM estimator """ expected_output = {'C2': 2, 'C1': 4, 'rect_kernel': 1, 'gamma': 0.625, 'kernel': 'RBF'} tsvm_cls = TSVM('RBF') tsvm_cls.set_params(**{'C1': 4, 'C2': 2, 'gamma': 0.625}) self.assertEqual(tsvm_cls.get_params(), expected_output, 'set_params and get_params output don\'t match')
def test_linear_tsvm_set_get_params(self): """ It checks that set_params and get_params works correctly for linear TSVM estimator """ expected_output = {'gamma': 1, 'C1': 0.5, 'rect_kernel': 1, 'C2': 1, 'kernel': 'linear'} tsvm_clf = TSVM('linear') tsvm_clf.set_params(**{'C1': 0.5, 'C2':1}) self.assertEqual(tsvm_clf.get_params(), expected_output, 'set_params and get_params output don\'t match')