def test_argmax(self):
v = BlockVector(3)
a = np.array([3, 2, 1])
v.set_block(0, a.copy())
v.set_block(1, a.copy())
v.set_block(2, a.copy())
v.get_block(1)[1] = 5
argmax = v.argmax()
self.assertEqual(argmax, 4)
def test_argmax(self):
v = BlockVector(2)
v[0] = np.arange(5)
v[1] = np.arange(10, 15)
self.assertEqual(v.argmax(), v.size-1)
