def test_factor4(self):
f = ['a','b','c']*10
fac = formula.factor('ff', set(f))
fac.namespace = {'ff':f}
m = fac.main_effect(reference=2)
r = N.array([N.identity(3)]*10)
r.shape = (30,3)
r = r.T
_m = N.array([r[0]-r[2],r[1]-r[2]])
assert_almost_equal(_m, m())
def test_factor2(self):
f = ['a','b','c']*10
fac = formula.factor('ff', set(f))
fac.namespace = {'ff':f}
self.assertEquals(fac().shape, (3,30))
def test_factor3(self):
f = ['a','b','c']*10
fac = formula.factor('ff', set(f))
fac.namespace = {'ff':f}
m = fac.main_effect(reference=1)
self.assertEquals(m().shape, (2,30))
def test_factor1(self):
f = ['a','b','c']*10
fac = formula.factor('ff', set(f))
fac.namespace = {'ff':f}
self.assertEquals(list(fac.values()), f)