def test01a(self): """Testing `resize()` method (decrease, large variation)""" b = ca.arange(self.N) b.resize(3) a = np.arange(3) #print "b->", `b` assert_array_equal(a, b[:], "Arrays are not equal")
def test01(self): """Testing `trim()` method (small chunklen)""" b = ca.arange(1e2, chunklen=2) b.trim(5) a = np.arange(1e2 - 5) #print "b->", `b` assert_array_equal(a, b[:], "Arrays are not equal")
def test02(self): """Testing `resize()` method (zero size)""" b = ca.arange(self.N) b.resize(0) a = np.arange(0) #print "b->", `b` assert_array_equal(a, b[:], "Arrays are not equal")
def test03(self): """Testing `trim()` method (complete trim)""" a = np.arange(0.) b = ca.arange(1e4) b.trim(1e4) #print "b->", `b` self.assert_(len(a) == len(b), "Lengths are not equal")
def test02(self): """Testing `trim()` method (large trim)""" a = np.arange(2) b = ca.arange(1e4) b.trim(1e4-2) #print "b->", `b` assert_array_equal(a, b[:], "Arrays are not equal")
def test01(self): """Testing `trim()` method (small chunklen)""" b = ca.arange(1e2, chunklen=2) b.trim(5) a = np.arange(1e2-5) #print "b->", `b` assert_array_equal(a, b[:], "Arrays are not equal")
def test05(self): """Testing `trim()` method (trimming zero items)""" a = np.arange(1e1) b = ca.arange(1e1) b.trim(0) #print "b->", `b` assert_array_equal(a, b[:], "Arrays are not equal")
def test00(self): """Testing `trim()` method""" b = ca.arange(1e3) b.trim(3) a = np.arange(1e3 - 3) #print "b->", `b` assert_array_equal(a, b[:], "Arrays are not equal")
def test01(self): """Testing evaluation of ndcarrays (int out)""" a = np.arange(np.prod(self.shape)).reshape(self.shape) b = ca.arange(np.prod(self.shape)).reshape(self.shape) outa = eval("a*2.+1") outb = ca.eval("b*2.+1") assert_array_equal(outa, outb, "Arrays are not equal")
def test02(self): """Testing `trim()` method (large trim)""" a = np.arange(2) b = ca.arange(1e4) b.trim(1e4 - 2) #print "b->", `b` assert_array_equal(a, b[:], "Arrays are not equal")
def test02b(self): """Testing `reshape()` (ndim -> ndim, II)""" a = np.arange(24, dtype="i4").reshape((2,3,4)) c = ca.arange(24, dtype="i4").reshape((4,3,2)) b = c.reshape((2,3,4)) #print "b->", `b` assert_array_equal(a, b, "Arrays are not equal")
def test00b(self): """Testing evaluation of ndcarrays (bool out, NumPy)""" a = np.arange(np.prod(self.shape)).reshape(self.shape) b = ca.arange(np.prod(self.shape)).reshape(self.shape) outa = eval("a>0") outb = ca.eval("b>0", out_flavor='numpy') assert_array_equal(outa, outb, "Arrays are not equal")
def test00(self): """Testing `trim()` method""" b = ca.arange(1e3) b.trim(3) a = np.arange(1e3-3) #print "b->", `b` assert_array_equal(a, b[:], "Arrays are not equal")
def test02b(self): """Testing `reshape()` (ndim -> ndim, II)""" a = np.arange(24, dtype="i4").reshape((2, 3, 4)) c = ca.arange(24, dtype="i4").reshape((4, 3, 2)) b = c.reshape((2, 3, 4)) #print "b->", `b` assert_array_equal(a, b, "Arrays are not equal")
def test00b(self): """Testing `carray` reshape (large shape)""" a = np.arange(16000).reshape((20,20,40)) b = ca.arange(16000, rootdir=self.rootdir).reshape((20,20,40)) if self.open: b = ca.open(rootdir=self.rootdir) #print "b->", `b` assert_array_equal(a, b, "Arrays are not equal")
def test00b(self): """Testing `resize()` method (increase)""" b = ca.arange(self.N) b.resize(self.N+3) a = np.arange(self.N+3) a[self.N:] = 0 #print "b->", `b` assert_array_equal(a, b[:], "Arrays are not equal")
def test01b(self): """Testing `resize()` method (increase, large variation)""" b = ca.arange(self.N, dflt=1) b.resize(self.N * 3) a = np.arange(self.N * 3) a[self.N:] = 1 #print "b->", `b` assert_array_equal(a, b[:], "Arrays are not equal")
def test01b(self): """Testing `resize()` method (increase, large variation)""" b = ca.arange(self.N, dflt=1) b.resize(self.N*3) a = np.arange(self.N*3) a[self.N:] = 1 #print "b->", `b` assert_array_equal(a, b[:], "Arrays are not equal")
def test06(self): """Testing `trim()` method (negative number of items)""" a = np.arange(2e1) b = ca.arange(1e1) b.trim(-10) a[10:] = 0 #print "b->", `b` assert_array_equal(a, b[:], "Arrays are not equal")
def test00b(self): """Testing `resize()` method (increase)""" b = ca.arange(self.N) b.resize(self.N + 3) a = np.arange(self.N + 3) a[self.N:] = 0 #print "b->", `b` assert_array_equal(a, b[:], "Arrays are not equal")
def test00a(self): """Testing `carray` reshape""" a = np.arange(16).reshape((2,2,4)) b = ca.arange(16, rootdir=self.rootdir).reshape((2,2,4)) if self.open: b = ca.open(rootdir=self.rootdir) #print "b->", `b` assert_array_equal(a, b, "Arrays are not equal")
def test02(self): """Testing iter() in combination with a list constructor""" b = ca.arange(self.N, dtype="f4") blist = list(b) blist1 = [r for r in b.iter(3,10)] self.assert_(blist1 == range(3,10)) blist2 = list(b) self.assert_(blist == blist2, "iter() not working correctly")
def test02(self): """Testing iter() in combination with a list constructor""" b = ca.arange(self.N, dtype="f4") blist = list(b) blist1 = [r for r in b.iter(3, 10)] self.assert_(blist1 == range(3, 10)) blist2 = list(b) self.assert_(blist == blist2, "where() not working correctly")
def test02(self): """Testing evaluation of ndcarrays (reduction)""" # Reduction ops are not supported yet a = np.arange(np.prod(self.shape)).reshape(self.shape) b = ca.arange(np.prod(self.shape)).reshape(self.shape) if ca.defaults.eval_vm: self.assertRaises(NotImplementedError, ca.eval, "sum(b)", depth=3) else: self.assertRaises(NotImplementedError, ca.eval, "b.sum(axis=0)", depth=3)
def test02(self): """Testing evaluation of ndcarrays (reduction, no axis)""" a = np.arange(np.prod(self.shape)).reshape(self.shape) b = ca.arange(np.prod(self.shape)).reshape(self.shape) if ca.defaults.eval_vm == "python": assert_array_equal(sum(a), ca.eval("sum(b)"), "Arrays are not equal") else: self.assertEqual(a.sum(), ca.eval("sum(b)"))
def test01a(self): """Testing where() in combination with a list constructor""" a = ca.zeros(self.N, dtype="bool") a[30:40] = ca.ones(10, dtype="bool") b = ca.arange(self.N, dtype="f4") blist = list(b) blist1 = [r for r in b.where(a)] self.assert_(blist1 == range(30,40)) blist2 = list(b) self.assert_(blist == blist2, "where() not working correctly")
def test01(self): """Testing where() in combination with a list constructor""" a = ca.zeros(self.N, dtype="bool") a[30:40] = ca.ones(10, dtype="bool") b = ca.arange(self.N, dtype="f4") blist = list(b) blist1 = [r for r in b.where(a)] self.assert_(blist1 == range(30, 40)) blist2 = list(b) self.assert_(blist == blist2, "where() not working correctly")
def test02b(self): """Testing evaluation of ndcarrays (reduction, with axis)""" a = np.arange(np.prod(self.shape)).reshape(self.shape) b = ca.arange(np.prod(self.shape)).reshape(self.shape) if ca.defaults.eval_vm == "python": # The Python VM does not have support for `axis` param assert_array_equal(sum(a), ca.eval("sum(b)"), "Arrays are not equal") else: assert_array_equal(a.sum(axis=1), ca.eval("sum(b, axis=1)"), "Arrays are not equal")
def test06(self): """Testing eval() with a mix of columns, numpy arrays and carrays""" N = 10 ra = np.fromiter(((i, i*2., i*3) for i in xrange(N)), dtype='i4,f8,i8') t = ca.ctable(ra) a = np.arange(N) b = ca.arange(N) ctr = t.eval("f0 + f1 - a + b") rar = ra['f0'] + ra['f1'] - a + b #print "ctable ->", ctr #print "numpy ->", rar assert_array_equal(ctr[:], rar, "ctable values are not correct")
def test06(self): """Testing eval() with a mix of columns, numpy arrays and carrays""" N = 10 ra = np.fromiter(((i, i * 2., i * 3) for i in xrange(N)), dtype='i4,f8,i8') t = ca.ctable(ra) a = np.arange(N) b = ca.arange(N) ctr = t.eval("f0 + f1 - a + b") rar = ra['f0'] + ra['f1'] - a + b #print "ctable ->", ctr #print "numpy ->", rar assert_array_equal(ctr[:], rar, "ctable values are not correct")
def test03(self): """Testing arange() with a `dtype`.""" a = np.arange(self.N, dtype="i1") ac = ca.arange(self.N, dtype="i1") self.assert_(np.all(a == ac))
import numpy as np import carray as ca from time import time N = 1e8 dtype = "i4" start, stop, step = 5, N, 4 t0 = time() a = np.arange(start, stop, step, dtype=dtype) print "Time numpy.arange() --> %.3f" % (time() - t0) t0 = time() ac = ca.arange(start, stop, step, dtype=dtype) print "Time carray.arange() --> %.3f" % (time() - t0) print "ac-->", ` ac ` # assert(np.all(a == ac))
def test04(self): """Testing `trim()` method (trimming more than available items)""" a = np.arange(0.) b = ca.arange(1e4) #print "b->", `b` self.assertRaises(ValueError, b.trim, 1e4+1)
import numpy as np import carray as ca from time import time N = 1e8 dtype = 'i4' start, stop, step = 5, N, 4 t0 = time() a = np.arange(start, stop, step, dtype=dtype) print "Time numpy.arange() --> %.3f" % (time()-t0) t0 = time() ac = ca.arange(start, stop, step, dtype=dtype) print "Time carray.arange() --> %.3f" % (time()-t0) print "ac-->", `ac` #assert(np.all(a == ac))
def test00(self): """Testing `carray` constructor""" a = np.arange(16).reshape((2, 2, 4)) b = ca.arange(16).reshape((2, 2, 4)) # print "b->", `b` assert_array_equal(a, b, "Arrays are not equal")
def test01b(self): """Testing where() with a multidimensional array""" a = ca.zeros((self.N, 10), dtype="bool") a[30:40] = ca.ones(10, dtype="bool") b = ca.arange(self.N*10, dtype="f4").reshape((self.N, 10)) self.assertRaises(NotImplementedError, b.where, a)
def test02(self): """Testing arange() with a `start`, `stop` and `step`.""" a = np.arange(3, self.N, 4) ac = ca.arange(3, self.N, 4) self.assert_(np.all(a == ac))
def test04(self): """Testing `trim()` method (trimming more than available items)""" a = np.arange(0.) b = ca.arange(1e4) #print "b->", `b` self.assertRaises(ValueError, b.trim, 1e4 + 1)
def test00(self): """Testing arange() with only a `stop`.""" a = np.arange(self.N) ac = ca.arange(self.N) self.assert_(np.all(a == ac))
def test00(self): """Testing `carray` constructor""" a = np.arange(16).reshape((2, 2, 4)) b = ca.arange(16).reshape((2, 2, 4)) #print "b->", `b` assert_array_equal(a, b, "Arrays are not equal")