def test_analysis_module(self):
rng = RandomNumberGenerator( )
module = self.createAnalysisModule()
ens_size = 12
obs_size = 1
state_size = 2
true_params = [1.25 , 0.75]
true_state = forward_model( true_params )
obs = [(measure( true_state ) , 0.75)]
A = Matrix( state_size , ens_size )
ens = []
for iens in range(ens_size):
param = [ random.gauss( 1.00 , 1.00 ) , random.gauss(1.00 , 1.00)]
ens.append( param )
mask = BoolVector(default_value = True , initial_size = ens_size)
mask[2] = False
(A , S , E , D , R , dObs) = init_matrices( ens , mask , obs , rng )
module.initUpdate( mask , S , R , dObs , E , D )
module.updateA( A , S , R , dObs , E , D )
mask[10] = False
mask[5] = False
(A , S , E , D , R , dObs) = init_matrices( ens , mask , obs , rng )
self.assertEqual( S.dims() , (obs_size , mask.countEqual( True )))
self.assertEqual( E.dims() , (obs_size , mask.countEqual( True )))
self.assertEqual( D.dims() , (obs_size , mask.countEqual( True )))
module.initUpdate( mask , S , R , dObs , E , D )
module.updateA( A , S , R , dObs , E , D )
def test_matrix_copy_column(self):
m = Matrix(10,2)
rng = RandomNumberGenerator(RngAlgTypeEnum.MZRAN, RngInitModeEnum.INIT_DEFAULT)
m.randomInit( rng )
with self.assertRaises(ValueError):
m.copyColumn(0,2)
with self.assertRaises(ValueError):
m.copyColumn(2,0)
with self.assertRaises(ValueError):
m.copyColumn(-2,0)
m.copyColumn(1, 0)
for i in range(m.rows()):
self.assertEqual( m[i,0] , m[i,1] )
def test_sub_copy(self):
m1 = Matrix(3,3)
rng = RandomNumberGenerator(RngAlgTypeEnum.MZRAN, RngInitModeEnum.INIT_DEFAULT)
m1.randomInit( rng )
with self.assertRaises(ValueError):
m2 = m1.subCopy( 0,0,4,2 )
with self.assertRaises(ValueError):
m2 = m1.subCopy( 0,0,2,4 )
with self.assertRaises(ValueError):
m2 = m1.subCopy( 4,0,1,1 )
with self.assertRaises(ValueError):
m2 = m1.subCopy( 0,2,1,2 )
m2 = m1.subCopy( 0,0,2,2 )
for i in range(2):
for j in range(2):
self.assertEqual( m1[i,j] , m2[i,j])
def createAnalysisModule(self):
rng = RandomNumberGenerator(RngAlgTypeEnum.MZRAN, RngInitModeEnum.INIT_DEFAULT)
return AnalysisModule(rng, lib_name = self.libname)
def setUp(self):
self.libname = ert.ert_lib_path + "/rml_enkf.so"
self.config_file = self.createTestPath(
"Statoil/config/obs_testing2/config")
self.rng = RandomNumberGenerator(RngAlgTypeEnum.MZRAN,
RngInitModeEnum.INIT_DEFAULT)
def test_matrix_random_init(self):
m = Matrix(10,10)
rng = RandomNumberGenerator(RngAlgTypeEnum.MZRAN, RngInitModeEnum.INIT_DEFAULT)
m.randomInit( rng )
grid.save_EGRID("grid/CASE.EGRID")
return grid
def make_field(rng, grid, iens):
permx = EclKW.create("PERMX", grid.getGlobalSize(),
EclTypeEnum.ECL_FLOAT_TYPE)
permx.assign(rng.getDouble())
poro = EclKW.create("PORO", grid.getGlobalSize(),
EclTypeEnum.ECL_FLOAT_TYPE)
poro.assign(rng.getDouble())
if not os.path.isdir("fields"):
os.makedirs("fields")
with open("fields/permx%d.grdecl" % iens, "w") as f:
permx.write_grdecl(f)
with open("fields/poro%d.grdecl" % iens, "w") as f:
poro.write_grdecl(f)
rng = RandomNumberGenerator()
rng.setState("ABCD6375ejascEFGHIJ")
grid = make_grid()
for iens in range(ens_size):
make_field(rng, grid, iens)
grid.save_EGRID("grid/CASE.EGRID")
return grid
def make_field(rng , grid , iens):
permx = EclKW.create( "PERMX" , grid.getGlobalSize( ) , EclTypeEnum.ECL_FLOAT_TYPE)
permx.assign( rng.getDouble( ) )
poro = EclKW.create( "PORO" , grid.getGlobalSize( ) , EclTypeEnum.ECL_FLOAT_TYPE)
poro.assign( rng.getDouble( ) )
if not os.path.isdir("fields"):
os.makedirs("fields")
with open("fields/permx%d.grdecl" % iens,"w") as f:
permx.write_grdecl( f )
with open("fields/poro%d.grdecl" % iens ,"w") as f:
poro.write_grdecl( f )
rng = RandomNumberGenerator( )
rng.setState("ABCD6375ejascEFGHIJ")
grid = make_grid( )
for iens in range(ens_size):
make_field(rng , grid , iens)