def test01(self): """Testing `resize()` (enlarge)""" a = np.ones((4, 3), dtype="i4") b = bcolz.ones((3, 3), dtype="i4", rootdir=self.rootdir) b.resize(4) # print "b->", `b` assert_array_equal(a, b, "Arrays are not equal")
def test01(self): """Testing `reshape()` (ndim -> unidim)""" a = np.ones(12, dtype="i4") c = bcolz.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 = bcolz.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 test02(self): """Testing `ones` constructor""" a = np.ones((2, 2), dtype='(4,)i4') b = bcolz.ones((2, 2), dtype='(4,)i4', rootdir=self.rootdir) if self.open: b = bcolz.open(rootdir=self.rootdir) # print "b->", `b` assert_array_equal(a, b, "Arrays are not equal")
def test02(self): """Testing unicode types (iter)""" a = np.ones((3,), dtype="U40") b = bcolz.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 test00(self): """Testing compound types (creation)""" a = np.ones((300, 4), dtype=self.dtype) b = bcolz.ones((300, 4), dtype=self.dtype) # print "b.dtype-->", b.dtype # print "b->", `b` self.assertTrue(a.dtype == b.dtype.base) assert_array_equal(a, b[:], "Arrays are not equal")
def test03(self): """Testing `reshape()` (0-dim)""" a = np.ones((0, 4), dtype="i4") b = bcolz.ones(0, dtype="i4").reshape((0, 4)) # print "b->", `b` # The next does not work well for carrays with shape (0,) # assert_array_equal(a, b, "Arrays are not equal") self.assertTrue(a.dtype.base == b.dtype.base) self.assertTrue(a.shape == b.shape + b.dtype.shape)
def test00a(self): """Testing wheretrue() in combination with a list constructor""" a = bcolz.zeros(self.N, dtype="bool", rootdir=self.rootdir) a[30:40] = bcolz.ones(10, dtype="bool") alist = list(a) blist1 = [r for r in a.wheretrue()] self.assertTrue(blist1 == list(range(30, 40))) alist2 = list(a) self.assertTrue(alist == alist2, "wheretrue() not working correctly")
def test00b(self): """Testing `resize()` (trim to zero)""" a = np.ones((0, 3), dtype="i4") b = bcolz.ones((3, 3), dtype="i4", rootdir=self.rootdir) b.resize(0) # print "b->", `b` # The next does not work well for carrays with shape (0,) # assert_array_equal(a, b, "Arrays are not equal") self.assertTrue("a.dtype.base == b.dtype.base") self.assertTrue("a.shape == b.shape+b.dtype.shape")
def test01a(self): """Testing where() in combination with a list constructor""" a = bcolz.zeros(self.N, dtype="bool") a[30:40] = bcolz.ones(10, dtype="bool") b = bcolz.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 is_in_ordered_subgroups(self, basket_col=None, bool_arr=None, _max_len_subgroup=1000): """""" assert basket_col is not None if bool_arr is None: bool_arr = bcolz.ones(self.len) return \ ctable_ext.is_in_ordered_subgroups( self[basket_col], bool_arr=bool_arr, _max_len_subgroup=_max_len_subgroup)
def test01(self): """Testing `iterblocks()` (w/ start, stop)""" a = np.ones((2, 3), dtype="i4") b = bcolz.ones((1000, 3), dtype="i4") # print "b->", `b` l, s = 0, 0 for block in bcolz.iterblocks(b, blen=2, start=10, stop=100): assert_array_equal(a, block, "Arrays are not equal") l += len(block) s += block.sum() self.assertEqual(l, 90) # as per Gauss summation formula self.assertEqual(s, 90 * 3)
def test01(self): """Testing `iterblocks()` (w/ start, stop)""" a = np.ones((2,3), dtype="i4") b = bcolz.ones((1000, 3), dtype="i4") # print "b->", `b` l, s = 0, 0 for block in bcolz.iterblocks(b, blen=2, start=10, stop=100): assert_array_equal(a, block, "Arrays are not equal") l += len(block) s += block.sum() self.assertEqual(l, 90) # as per Gauss summation formula self.assertEqual(s, 90*3)
def test00(self): """Testing `iterblocks()` (no start, stop, step)""" N = 1000 a = np.ones((2,3), dtype="i4") b = bcolz.ones((N, 3), dtype="i4") # print "b->", `b` l, s = 0, 0 for block in bcolz.iterblocks(b, blen=2): assert_array_equal(a, block, "Arrays are not equal") l += len(block) s += block.sum() self.assertEqual(l, N) # as per Gauss summation formula self.assertEqual(s, N*3)
def test00(self): """Testing `iterblocks()` (no start, stop, step)""" N = 1000 a = np.ones((2, 3), dtype="i4") b = bcolz.ones((N, 3), dtype="i4") # print "b->", `b` l, s = 0, 0 for block in bcolz.iterblocks(b, blen=2): assert_array_equal(a, block, "Arrays are not equal") l += len(block) s += block.sum() self.assertEqual(l, N) # as per Gauss summation formula self.assertEqual(s, N * 3)
def test01b(self): """Testing where() with a multidimensional array""" a = bcolz.zeros((self.N, 10), dtype="bool", rootdir=self.rootdir) a[30:40] = bcolz.ones(10, dtype="bool") b = bcolz.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 = bcolz.zeros((self.N, 10), dtype="bool", rootdir=self.rootdir) a[30:40] = bcolz.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 = bcolz.ones(12, dtype="i4").reshape((3, -1)) # print "b->", `b` assert_array_equal(a, b, "Arrays are not equal")