コード例 #1
0
ファイル: test_rml.py プロジェクト: blattms/ert-debian
def init_matrices(ens, mask, obs, rng):
    state_size = 2
    report_step = 5
    meas_data = MeasData(mask)
    meas_block = meas_data.addBlock("OBS", report_step, len(obs))

    A = Matrix(state_size, mask.countEqual(True))
    active_iens = 0
    for iens, params in enumerate(ens):
        if mask[iens]:
            state = forward_model(params)
            meas_block[0, iens] = measure(state)

            A[0, active_iens] = params[0]
            A[1, active_iens] = params[1]

            active_iens += 1

    S = meas_data.createS()

    obs_data = ObsData()
    obs_block = obs_data.addBlock("OBS", 1)
    for iobs, obs_value in enumerate(obs):
        obs_block[iobs] = obs_value

    R = obs_data.createR()
    dObs = obs_data.createDObs()
    E = obs_data.createE(rng, meas_data.getActiveEnsSize())
    D = obs_data.createD(E, S)

    obs_data.scale(S, E=E, D=D, R=R, D_obs=dObs)
    return (A, S, E, D, R, dObs)
コード例 #2
0
    def test_create(self):
        obs_data = ObsData()
        obs_size = 10
        block = obs_data.addBlock("OBS", obs_size)
        self.assertTrue(isinstance(block, ObsBlock))

        block[0] = (10, 10)
        block[1] = (12, 12)
        D = obs_data.createDObs()
        self.assertTrue(isinstance(D, Matrix))
        self.assertEqual(D.dims(), (2, 1))

        self.assertEqual(D[0, 0], 10)
        self.assertEqual(D[1, 0], 12)

        obs_data.scaleMatrix(D)
        self.assertEqual(D[0, 0], 1)
        self.assertEqual(D[1, 0], 1)

        R = obs_data.createR()
        self.assertEqual((2, 2), R.dims())

        with self.assertRaises(IndexError):
            obs_data[10]

        v, s = obs_data[0]
        self.assertEqual(v, 10)
        self.assertEqual(s, 10)

        v, s = obs_data[1]
        self.assertEqual(v, 12)
        self.assertEqual(s, 12)
コード例 #3
0
ファイル: test_rml.py プロジェクト: YingfangZhou/ert
def init_matrices(ens , mask , obs , rng):
    state_size = 2
    report_step = 5
    meas_data = MeasData( mask )
    meas_block = meas_data.addBlock("OBS" , report_step , len(obs) )

    A = Matrix( state_size , mask.countEqual( True ))
    active_iens = 0
    for iens,params in enumerate( ens ):
        if mask[iens]:
            state = forward_model( params )
            meas_block[0,iens] = measure( state )
            
            A[0 , active_iens] = params[0]
            A[1 , active_iens] = params[1]
            
            active_iens += 1

            
    S = meas_data.createS()
        
    obs_data = ObsData()
    obs_block = obs_data.addBlock("OBS" , 1)
    for iobs,obs_value in enumerate(obs):
        obs_block[iobs] = obs_value
    

    R = obs_data.createR()
    dObs = obs_data.createDObs()
    E = obs_data.createE( rng , meas_data.getActiveEnsSize() )
    D = obs_data.createD(E , S)
        
    obs_data.scale(S , E = E , D = D , R = R , D_obs = dObs)
    return (A , S , E , D , R , dObs)
コード例 #4
0
ファイル: test_obs_data.py プロジェクト: akva2/ert
    def test_create(self):
        obs_data = ObsData()
        obs_size = 10
        block = obs_data.addBlock("OBS" , obs_size)
        self.assertTrue( isinstance( block , ObsBlock ))

        block[0] = (10,10)
        block[1] = (12,12)
        D = obs_data.createDObs()
        self.assertTrue( isinstance(D , Matrix ))
        self.assertEqual( D.dims() , (2,1))
        
        self.assertEqual( D[0,0] , 10 )
        self.assertEqual( D[1,0] , 12 )
        
        obs_data.scaleMatrix( D )
        self.assertEqual( D[0,0] , 1 )
        self.assertEqual( D[1,0] , 1 )
        
        R = obs_data.createR()
        self.assertEqual( (2,2) , R.dims() )

        with self.assertRaises(IndexError):
            obs_data[10]

        v,s = obs_data[0]
        self.assertEqual( v , 10 )
        self.assertEqual( s , 10 )


        v,s = obs_data[1]
        self.assertEqual( v , 12 )
        self.assertEqual( s , 12 )