def test01(self):
"""Testing `reshape()` (ndim -> unidim)"""
a = np.ones(12, dtype="i4")
c = blz.ones(12, dtype="i4").reshape((3,4))
b = c.reshape(12)
#print "b->", `b`
assert_array_equal(a, b, "Arrays are not equal")
def test02(self):
"""Testing `iter()` (w/ start, stop, step)"""
a = np.ones((3,), dtype="i4")
b = blz.ones((1000,3), dtype="i4")
#print "b->", `b`
for r in b.iter(15, 100, 3):
assert_array_equal(a, r, "Arrays are not equal")
def test01(self):
"""Testing `resize()` (enlarge)"""
a = np.ones((4,3), dtype="i4")
b = blz.ones((3,3), dtype="i4", rootdir=self.rootdir)
b.resize(4)
#print "b->", `b`
assert_array_equal(a, b, "Arrays are not equal")
def test02c(self):
"""Testing ones() constructor, with a string type"""
a = np.ones(self.N, dtype='S3')
ac = blz.ones(self.N, dtype='S3', rootdir=self.rootdir)
#print "a-->", a, ac
self.assert_(a.dtype == ac.dtype)
self.assert_(np.all(a == ac[:]))
def test02(self):
"""Testing unicode types (iter)"""
a = np.ones((3,), dtype="U40")
b = blz.ones((1000,3), dtype="U40")
#print "b->", `b`
for r in b.iter():
#print "r-->", r
assert_array_equal(a, r, "Arrays are not equal")
def test02(self):
"""Testing `ones` constructor"""
a = np.ones((2,2), dtype='(4,)i4')
b = blz.ones((2,2), dtype='(4,)i4', rootdir=self.rootdir)
if self.open:
b = blz.open(rootdir=self.rootdir)
#print "b->", `b`
assert_array_equal(a, b, "Arrays are not equal")
def test00(self):
"""Testing compound types (creation)"""
a = np.ones((300,4), dtype=self.dtype)
b = blz.ones((300,4), dtype=self.dtype)
#print "b.dtype-->", b.dtype
#print "b->", `b`
self.assert_(a.dtype == b.dtype.base)
assert_array_equal(a, b[:], "Arrays are not equal")
def test00a(self):
"""Testing wheretrue() in combination with a list constructor"""
a = blz.zeros(self.N, dtype="bool")
a[30:40] = blz.ones(10, dtype="bool")
alist = list(a)
blist1 = [r for r in a.wheretrue()]
self.assert_(blist1 == list(range(30,40)))
alist2 = list(a)
self.assert_(alist == alist2, "wheretrue() not working correctly")
def test03(self):
"""Testing `reshape()` (0-dim)"""
a = np.ones((0,4), dtype="i4")
b = blz.ones(0, dtype="i4").reshape((0,4))
#print "b->", `b`
# The next does not work well for barrays with shape (0,)
#assert_array_equal(a, b, "Arrays are not equal")
self.assert_(a.dtype.base == b.dtype.base)
self.assert_(a.shape == b.shape+b.dtype.shape)
def test00b(self):
"""Testing `resize()` (trim to zero)"""
a = np.ones((0,3), dtype="i4")
b = blz.ones((3,3), dtype="i4", rootdir=self.rootdir)
b.resize(0)
#print "b->", `b`
# The next does not work well for barrays with shape (0,)
#assert_array_equal(a, b, "Arrays are not equal")
self.assert_("a.dtype.base == b.dtype.base")
self.assert_("a.shape == b.shape+b.dtype.shape")
def test01b(self):
"""Testing where() with a multidimensional array"""
a = blz.zeros((self.N, 10), dtype="bool")
a[30:40] = blz.ones(10, dtype="bool")
b = blz.arange(self.N*10, dtype="f4").reshape((self.N, 10))
self.assertRaises(NotImplementedError, b.where, a)
def test00b(self):
"""Testing wheretrue() with a multidimensional array"""
a = blz.zeros((self.N, 10), dtype="bool")
a[30:40] = blz.ones(10, dtype="bool")
self.assertRaises(NotImplementedError, a.wheretrue)
def test00c(self):
"""Testing `reshape()` (unidim -> ndim, -1 in newshape (II))"""
a = np.ones((3,4), dtype="i4")
b = blz.ones(12, dtype="i4").reshape((3,-1))
#print "b->", `b`
assert_array_equal(a, b, "Arrays are not equal")
def test02b(self):
"""Testing ones() constructor, with a `dtype`."""
a = np.ones(self.N, dtype='i4')
ac = blz.ones(self.N, dtype='i4', rootdir=self.rootdir)
self.assert_(a.dtype == ac.dtype)
self.assert_(np.all(a == ac[:]))
def test02a(self):
"""Testing ones() constructor."""
a = np.ones(self.N)
ac = blz.ones(self.N, rootdir=self.rootdir)
self.assert_(a.dtype == ac.dtype)
self.assert_(np.all(a == ac[:]))