def test_glmnet_c(): # define binary prob data = datasets['dumb2'] # use GLMNET on binary problem clf = GLMNET_C() clf.ca.enable('estimates') clf.train(data) # test predictions pre = clf.predict(data.samples) assert_array_equal(pre, data.targets)