示例#1
0
 def testArray(self):
     f = ufunc_mod.BinaryCallable()
     a = numpy.random.randn(5)
     b = numpy.random.randn(5)
     assert_array_almost_equal(f(a,b), (a*2+b*3)) 
     c = numpy.zeros(5, dtype=float)
     d = f(a,b,output=c)
     self.assert_(c is d)
     assert_array_almost_equal(d, a*2 + b*3) 
     assert_array_almost_equal(f(a, 2.0), a*2 + 6.0) 
     assert_array_almost_equal(f(1.0, b), 2.0 + b*3) 
示例#2
0
 def testScalar(self):
     f = ufunc_mod.BinaryCallable()
     assert_array_almost_equal(f(1.0, 3.0), 11.0) 
     assert_array_almost_equal(f(3.0, 2.0), 12.0)