def test_clip(self):
v = BlockVector(3)
v2 = BlockVector(3)
a = np.zeros(5)
b = np.ones(3)*5.0
c = np.ones(3)*10.0
v[0] = a
v[1] = b
v[2] = c
v2[0] = np.ones(5) * 4.0
v2[1] = np.ones(3) * 5.0
v2[2] = np.ones(3) * 9.0
vv = v.clip(4.0, 9.0)
self.assertEqual(vv.nblocks, v.nblocks)
for bid, blk in enumerate(vv):
self.assertTrue(np.allclose(blk, v2[bid]))
with self.assertRaises(NotImplementedError) as ctx:
vv = v.clip(4.0, 9.0, out=v2)
def test_clip(self):
v = BlockVector(3)
v2 = BlockVector(3)
a = np.zeros(5)
b = np.ones(3) * 5.0
c = np.ones(3) * 10.0
v.set_block(0, a)
v.set_block(1, b)
v.set_block(2, c)
v2.set_block(0, np.ones(5) * 4.0)
v2.set_block(1, np.ones(3) * 5.0)
v2.set_block(2, np.ones(3) * 9.0)
vv = v.clip(4.0, 9.0)
self.assertEqual(vv.nblocks, v.nblocks)
for bid, blk in enumerate(vv):
self.assertTrue(np.allclose(blk, v2.get_block(bid)))
