def test_le(self): v = BlockVector(2) a = np.ones(5) b = np.zeros(9) v[0] = a v[1] = b flags = v <= 1 v[1] = b + 1 self.assertEqual(v.nblocks, flags.nblocks) for bid, blk in enumerate(flags): self.assertTrue(np.allclose(blk, v[bid])) flags = v <= v vv = v.copy() vv.fill(1.0) self.assertEqual(v.nblocks, flags.nblocks) for bid, blk in enumerate(flags): self.assertTrue(np.allclose(blk, vv[bid])) flags = v <= v.flatten() vv = v.copy() vv.fill(1.0) self.assertEqual(v.nblocks, flags.nblocks) for bid, blk in enumerate(flags): self.assertTrue(np.allclose(blk, vv[bid]))
def test_le(self): v = BlockVector(2) a = np.ones(5) b = np.zeros(9) v.set_block(0, a) v.set_block(1, b) flags = v <= 1 v.set_block(1, b + 1) self.assertEqual(v.nblocks, flags.nblocks) for bid, blk in enumerate(flags): self.assertTrue(np.allclose(blk, v.get_block(bid))) flags = v <= v vv = v.copy() vv.fill(1.0) self.assertEqual(v.nblocks, flags.nblocks) for bid, blk in enumerate(flags): self.assertTrue(np.allclose(blk, vv.get_block(bid))) flags = v <= v.flatten() vv = v.copy() vv.fill(1.0) self.assertEqual(v.nblocks, flags.nblocks) for bid, blk in enumerate(flags): self.assertTrue(np.allclose(blk, vv.get_block(bid)))
def test_ge(self): v = BlockVector(2) a = np.ones(5) b = np.zeros(9) v.set_block(0, a) v.set_block(1, b) flags = v >= 0 v.set_block(1, b + 1) self.assertEqual(v.nblocks, flags.nblocks) for bid, blk in enumerate(flags): self.assertTrue(np.allclose(blk, v.get_block(bid))) v.set_block(1, b - 1) flags = v >= np.zeros(v.size) v.set_block(1, b) self.assertEqual(v.nblocks, flags.nblocks) for bid, blk in enumerate(flags): self.assertTrue(np.allclose(blk, v.get_block(bid))) v.set_block(1, b - 1) vv = v.copy() vv.fill(0.0) flags = v >= vv v.set_block(1, b) self.assertEqual(v.nblocks, flags.nblocks) for bid, blk in enumerate(flags): self.assertTrue(np.allclose(blk, v.get_block(bid)))
def test_lt(self): v = BlockVector(2) a = np.ones(5) b = np.zeros(9) v[0] = a v[1] = b flags = v < 1 v[0] = a-1 v[1] = b+1 self.assertEqual(v.nblocks, flags.nblocks) for bid, blk in enumerate(flags): self.assertTrue(np.allclose(blk, v[bid])) v[0] = a + 1 v[1] = b - 1 flags = v < np.ones(v.size) v[0] = a - 1 v[1] = b + 1 self.assertEqual(v.nblocks, flags.nblocks) for bid, blk in enumerate(flags): self.assertTrue(np.allclose(blk, v[bid])) v[0] = a + 1 v[1] = b - 1 vv = v.copy() vv.fill(1.0) flags = v < vv v[0] = a - 1 v[1] = b + 1 self.assertEqual(v.nblocks, flags.nblocks) for bid, blk in enumerate(flags): self.assertTrue(np.allclose(blk, v[bid]))
def test_copy(self): v = BlockVector(2) a = np.ones(5) b = np.zeros(9) v.set_block(0, a) v.set_block(1, b) v2 = v.copy() self.assertTrue(np.allclose(v.flatten(), v2.flatten()))
def test_copy(self): v = BlockVector(2) a = np.ones(5) b = np.zeros(9) v[0] = a v[1] = b v2 = v.copy() self.assertTrue(np.allclose(v.flatten(), v2.flatten()))
def test_ne(self): v = BlockVector(2) a = np.ones(5) b = np.zeros(9) v[0] = a v[1] = b flags = v != 0 self.assertEqual(v.nblocks, flags.nblocks) for bid, blk in enumerate(flags): self.assertTrue(np.allclose(blk, v[bid])) flags = v != np.zeros(v.size) self.assertEqual(v.nblocks, flags.nblocks) for bid, blk in enumerate(flags): self.assertTrue(np.allclose(blk, v[bid])) vv = v.copy() vv.fill(0.0) flags = v != vv self.assertEqual(v.nblocks, flags.nblocks) for bid, blk in enumerate(flags): self.assertTrue(np.allclose(blk, v[bid]))