Python BlockVector.argmaxの例

コード例 #1

ファイル: test_block_vector.py プロジェクト: vova292/pyomo

 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)

コード例 #2

ファイル: test_block_vector.py プロジェクト: CanLi1/pyomo-1

 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)