added build mode tests to compile.py, which fail

上级 3c371923
...@@ -210,12 +210,33 @@ class _test_single(unittest.TestCase): ...@@ -210,12 +210,33 @@ class _test_single(unittest.TestCase):
core.pop_mode() core.pop_mode()
def test_3(self): def test_3(self):
a = core.Numpy2(data=numpy.ones((2,2))) a = core.Numpy2(data=numpy.random.rand(2,2))
b = core.Numpy2(data=numpy.ones((2,2))) b = core.Numpy2(data=numpy.random.rand(2,2))
c = core.add(a,b) c = core.add(a,b)
self.failUnless(c.data is None)
self.failUnless(c.state is Empty)
new_a = numpy.random.rand(2,2)
new_b = numpy.random.rand(2,2)
a.data = new_a
b.data = new_b
p = single(c) p = single(c)
p() p()
self.failUnless(core._approx_eq(c, numpy.ones((2,2))*2)) self.failUnless(core._approx_eq(c, new_a + new_b))
def test_get_element(self):
a_data = numpy.random.rand(2,2)
a = core.Numpy2(data=a_data)
a_i = a[0,0]
p = single(a_i)
for i in 0,1:
for j in 0,1:
p()
self.failUnless(a_data[i,j] == a_i.data)
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() unittest.main()
......
...@@ -1729,7 +1729,7 @@ class _testCase_slicing(unittest.TestCase): ...@@ -1729,7 +1729,7 @@ class _testCase_slicing(unittest.TestCase):
wa1 = wrap(a)[0:8:2] wa1 = wrap(a)[0:8:2]
for i in xrange(8): a[i] = i for i in xrange(8): a[i] = i
self.failUnless(wa1.data.shape == (4,)) self.failUnless(wa1.shape == (4,))
for i in xrange(4): for i in xrange(4):
self.failUnless(a[i*2] == wa1.data[i]) self.failUnless(a[i*2] == wa1.data[i])
def test_getslice_3d_float(self): def test_getslice_3d_float(self):
...@@ -1737,10 +1737,18 @@ class _testCase_slicing(unittest.TestCase): ...@@ -1737,10 +1737,18 @@ class _testCase_slicing(unittest.TestCase):
a = numpy.asarray(range(4*5*6)) a = numpy.asarray(range(4*5*6))
a.resize((4,5,6)) a.resize((4,5,6))
wa1 = wrap(a)[1:3] wa1 = wrap(a)[1:3]
wa1.data.shape self.failUnless(wa1.shape == (2,5,6))
self.failUnless(numpy.all(a[1:3] == wa1.data)) self.failUnless(numpy.all(a[1:3] == wa1.data))
a[1] *= -1.0 a[1] *= -1.0
self.failUnless(numpy.all(a[1:3] == wa1.data)) self.failUnless(numpy.all(a[1:3] == wa1.data))
def test_getslice_3d_one(self):
"""Test getslice on 3d array"""
a = numpy.asarray(range(4*5*6))
a.resize((4,5,6))
wa = wrap(a)
wa_123 = wa[1,2,3]
self.failUnless(wa_123.shape == (), wa_123.shape)
add = scalar_switch(add_elemwise, add_scalar, add_scalar) add = scalar_switch(add_elemwise, add_scalar, add_scalar)
add_inplace = scalar_switch(add_elemwise_inplace, add_scalar_inplace) add_inplace = scalar_switch(add_elemwise_inplace, add_scalar_inplace)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论