def test_termcolumns(self):
t1 = formula.term("A")
t2 = formula.term("B")
f = t1 + t2 + t1 * t2
def other(val):
return N.array([3.2*val,4.342*val**2, 5.234*val**3])
q = formula.quantitative(['other%d' % i for i in range(1,4)], termname='other', func=t1, transform=other)
f += q
q.namespace = f.namespace = self.formula.namespace
assert_almost_equal(q(), f()[f.termcolumns(q)])
def test_quantitative(self):
t = self.terms[2]
sint = formula.quantitative('t', func=t, transform=N.sin)
t.namespace = sint.namespace = self.formula.namespace
assert_almost_equal(N.sin(t()), sint())
