# -*- coding: utf-8 -*-
"""
Created on Sat Apr 26 11:21:07 2014

Author: josef
"""
import numpy as np
from numpy.testing import assert_almost_equal

import statsmodels.api as sm
from statsmodels.discrete.tests.results.results_discrete import Spector


hist = []
def callb(*args):
    hist.append(args[0])

data = sm.datasets.spector.load()
data.exog = sm.add_constant(data.exog, prepend=False)
res2 = Spector()
res2.probit()
cls_res2 = res2
cls_res1 = sm.Probit(data.endog, data.exog).fit(method="cg",
    disp=1, maxiter=1000, gtol=1e-08, callback=callb)
    
print cls_res1.params
print cls_res2.params
print cls_res1.params / cls_res2.params - 1
print cls_res1.params - cls_res2.params

print cls_res1.mle_retvals

import pandas
res_ = pandas.DataFrame(np.asarray(hist))
print res_.to_string()
#res_.to_csv('trace_probitcg_32.csv')
#res_.to_csv('trace_probitcg_64wheel.csv')

DECIMAL_4 = 4
assert_almost_equal(cls_res1.params, cls_res2.params, DECIMAL_4)