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')