提交 cdeeb509 authored 作者: Frederic's avatar Frederic

use numpy.allclose in a test.

上级 fecb951f
...@@ -61,9 +61,9 @@ def test_pycuda_elemwise_source_module(): ...@@ -61,9 +61,9 @@ def test_pycuda_elemwise_source_module():
val1 = numpy.asarray(numpy.random.rand(*shape), dtype='float32') val1 = numpy.asarray(numpy.random.rand(*shape), dtype='float32')
val2 = numpy.asarray(numpy.random.rand(*shape), dtype='float32') val2 = numpy.asarray(numpy.random.rand(*shape), dtype='float32')
assert (f(val1, val2) == f2(val1, val2)).all() assert numpy.allclose(f(val1, val2), f2(val1, val2))
assert (f(val1, val2) == f3(val1, val2)).all() assert numpy.allclose(f(val1, val2), f3(val1, val2))
assert (f(val1, val2) == f4(val1, val2)).all() assert numpy.allclose(f(val1, val2), f4(val1, val2))
#print f(val1,val2) #print f(val1,val2)
#print f2(val1,val2) #print f2(val1,val2)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论