def test_argmin(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 argmin = v.argmin() self.assertEqual(argmin, 4)
def test_argmin(self): v = BlockVector(2) v[0] = np.arange(5) v[1] = np.arange(10, 15) self.assertEqual(v.argmin(), 0)