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)