1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34
|
"""
Test functions for models.glm
"""
import numpy as N
import numpy.random as R
from numpy.testing import NumpyTest, NumpyTestCase
import scipy.sandbox.models as S
import scipy.sandbox.models.glm as models
W = R.standard_normal
class test_Regression(NumpyTestCase):
def check_Logistic(self):
X = W((40,10))
Y = N.greater(W((40,)), 0)
family = S.family.Binomial()
cmodel = models(design=X, family=S.family.Binomial())
results = cmodel.fit(Y)
self.assertEquals(results.df_resid, 30)
def check_Logisticdegenerate(self):
X = W((40,10))
X[:,0] = X[:,1] + X[:,2]
Y = N.greater(W((40,)), 0)
family = S.family.Binomial()
cmodel = models(design=X, family=S.family.Binomial())
results = cmodel.fit(Y)
self.assertEquals(results.df_resid, 31)
if __name__ == "__main__":
NumpyTest().run()
|