def testmrbdclcovKnownValues(self):
"""mrbdcl/cov should give known result with known input"""
for mypath, itype, iunit in self.knownValues:
rmn.mrfopt(rmn.FSTOP_MSGLVL, rmn.BURPOP_MSG_FATAL)
funit = rmn.burp_open(self.getFN(mypath))
nbrp = rmn.mrfnbr(funit)
maxlen = max(64, rmn.mrfmxl(funit)) + 10
buf = None
handle = 0
handle = rmn.mrfloc(funit, handle)
buf = rmn.mrfget(handle, buf, funit)
params = rmn.mrbhdr(buf)
for iblk in range(params['nblk']):
blkparams = rmn.mrbprm(buf, iblk + 1)
blkdata = rmn.mrbxtr(buf, iblk + 1)
lstelebufr = rmn.mrbdcl(blkdata['cmcids'])
rmn.burp_close(funit)
lstelebufr0 = _np.array([
7004, 11001, 11002, 12001, 12192, 10194, 8001, 11003, 11004,
13210
],
dtype=_np.int32)
self.assertFalse(_np.any(lstelebufr0 - lstelebufr != 0))
lstelecmc = rmn.mrbcol(lstelebufr)
self.assertFalse(_np.any(lstelecmc - blkdata['cmcids'] != 0))
v1 = rmn.mrbdcl(blkdata['cmcids'][0])
v2 = rmn.mrbcol(v1)
self.assertEqual(lstelebufr0[0], v1)
self.assertEqual(blkdata['cmcids'][0], v2)
def testmrbtypKnownValues(self):
"""mrbtyp should give known result with known input"""
for mypath, itype, iunit in self.knownValues:
rmn.mrfopt(rmn.FSTOP_MSGLVL, rmn.BURPOP_MSG_FATAL)
funit = rmn.burp_open(self.getFN(mypath))
buf = None
handle = 0
handle = rmn.mrfloc(funit, handle)
buf = rmn.mrfget(handle, buf, funit)
params = rmn.mrbhdr(buf)
for iblk in range(params['nblk']):
blkparams = rmn.mrbprm(buf, iblk + 1)
blktypdict = rmn.mrbtyp_decode(blkparams['btyp'])
btyp1 = rmn.mrbtyp_encode(blktypdict)
btyp2 = rmn.mrbtyp_encode(blktypdict['bknat'],
blktypdict['bktyp'],
blktypdict['bkstp'])
bknat = rmn.mrbtyp_encode_bknat(blkparams['bknat_multi'],
blkparams['bknat_kind'])
bktyp = rmn.mrbtyp_encode_bktyp(blkparams['bktyp_alt'],
blkparams['bktyp_kind'])
btyp3 = rmn.mrbtyp_encode(bknat, bktyp, blkparams['bkstp'])
self.assertEqual(btyp1, blkparams['btyp'])
self.assertEqual(btyp2, blkparams['btyp'])
self.assertEqual(blkparams['bknat'], bknat)
self.assertEqual(blkparams['bktyp'], bktyp)
self.assertEqual(btyp3, blkparams['btyp'])
rmn.burp_close(funit)
def testmrbcvtencodeKnownValues(self):
"""mrbprm should give known result with known input"""
for mypath, itype, iunit in self.knownValues:
rmn.mrfopt(rmn.FSTOP_MSGLVL, rmn.BURPOP_MSG_FATAL)
funit = rmn.burp_open(self.getFN(mypath))
nbrp = rmn.mrfnbr(funit)
maxlen = max(64, rmn.mrfmxl(funit)) + 10
buf = None
handle = 0
handle = rmn.mrfloc(funit, handle)
buf = rmn.mrfget(handle, buf, funit)
params = rmn.mrbhdr(buf)
for iblk in range(params['nblk']):
blkparams = rmn.mrbprm(buf, iblk + 1)
blkdata = rmn.mrbxtr(buf, iblk + 1)
rval = rmn.mrbcvt_decode(blkdata['cmcids'], blkdata['tblval'],
blkparams['datyp'])
tblval = rmn.mrbcvt_encode(blkdata['cmcids'], rval)
for iele in range(blkdata['cmcids'].size):
e_cmcids = blkdata['cmcids'][iele]
e_cmcids = rmn.mrbcvt_dict_bufr(
e_cmcids, raise_error=False)['e_bufrid']
e_rval = rval[iele, :, :]
e_tblval0 = blkdata['tblval'][iele, :, :]
e_tblval1 = tblval[iele, :, :]
if not _np.all(e_tblval0 == e_tblval1):
## print 'b1',repr(blkparams)
print('b2', repr(blkdata))
print('b3', repr(rmn.mrbcvt_dict(e_cmcids)))
print(
'b4',
repr(
rmn.mrbcvt_dict_bufr(e_cmcids,
raise_error=False)))
print('rf', repr(e_rval.ravel()), blkparams['datyp'])
print(
'rv',
repr(_np.round(e_rval).astype(_np.int32).ravel()),
blkparams['datyp'])
print('t0', repr(e_tblval0.ravel()))
print('t1', repr(e_tblval1.ravel()))
#TODO: problem w/ values < 0: decode int as is... but encode shift values by -1
self.assertTrue(
_np.all(e_tblval0 == e_tblval1),
"{}, {}: id={}, \nrval:{}, \nexp:{}, \ngot:{}".format(
iblk, iele, e_cmcids, e_rval.ravel(),
e_tblval0.ravel(), e_tblval1.ravel()))
rmn.burp_close(funit)
def testmrbprmKnownValues(self):
"""mrbprm should give known result with known input"""
for mypath, itype, iunit in self.knownValues:
rmn.mrfopt(rmn.FSTOP_MSGLVL, rmn.BURPOP_MSG_FATAL)
funit = rmn.burp_open(self.getFN(mypath))
nbrp = rmn.mrfnbr(funit)
maxlen = max(64, rmn.mrfmxl(funit)) + 10
buf = None
handle = 0
handle = rmn.mrfloc(funit, handle)
buf = rmn.mrfget(handle, buf, funit)
params = rmn.mrbhdr(buf)
for iblk in range(params['nblk']):
blkparams = rmn.mrbprm(buf, iblk + 1)
blkparams0 = {
'datypd': 'uint',
'nele': 10,
'nbit': 20,
'datyp': 2,
'nval': 17,
'bdesc': 0,
'btyp': 9326,
'bfam': 10,
'nt': 1,
'bit0': 288,
'bktyp': 70,
'bkstp': 14,
'bknat': 4,
'bknat_multi': 1,
'bknat_kind': 0,
'bknat_kindd': 'data',
'bktyp_alt': 1,
'bktyp_kind': 6,
'bkno': iblk + 1,
'bktyp_kindd': 'data seen by OA at altitude, global model',
'bkstpd': "statistiques d'erreur d'observation"
}
## print 1,blkparams
for k in blkparams.keys():
self.assertEqual(
blkparams0[k], blkparams[k],
'For {0}, expected {1}, got {2}'.format(
k, blkparams0[k], blkparams[k]))
rmn.burp_close(funit)
def testmrbcvtdecodeKnownValues(self):
"""mrbprm should give known result with known input"""
for mypath, itype, iunit in self.knownValues:
rmn.mrfopt(rmn.FSTOP_MSGLVL, rmn.BURPOP_MSG_FATAL)
funit = rmn.burp_open(self.getFN(mypath))
nbrp = rmn.mrfnbr(funit)
maxlen = max(64, rmn.mrfmxl(funit)) + 10
buf = None
handle = 0
handle = rmn.mrfloc(funit, handle)
buf = rmn.mrfget(handle, buf, funit)
params = rmn.mrbhdr(buf)
for iblk in range(params['nblk']):
blkparams = rmn.mrbprm(buf, iblk + 1)
blkdata = rmn.mrbxtr(buf, iblk + 1)
rval = rmn.mrbcvt_decode(blkdata['cmcids'], blkdata['tblval'],
blkparams['datyp'])
rval = rmn.mrbcvt_decode(blkdata, datyp=blkparams['datyp'])
#TODO: check results
rmn.burp_close(funit)
def testmrbxtrKnownValues(self):
"""mrbprm should give known result with known input"""
for mypath, itype, iunit in self.knownValues:
rmn.mrfopt(rmn.FSTOP_MSGLVL, rmn.BURPOP_MSG_FATAL)
funit = rmn.burp_open(self.getFN(mypath))
nbrp = rmn.mrfnbr(funit)
maxlen = max(64, rmn.mrfmxl(funit)) + 10
buf = None
handle = 0
handle = rmn.mrfloc(funit, handle)
buf = rmn.mrfget(handle, buf, funit)
params = rmn.mrbhdr(buf)
## blkdata = {
## 'cmcids' : None,
## 'tblval' : None
## }
for iblk in range(params['nblk']):
blkparams = rmn.mrbprm(buf, iblk + 1)
## blkdata = rmn.mrbxtr(buf, iblk+1, blkdata['cmcids'], blkdata['tblval'])
blkdata = rmn.mrbxtr(buf, iblk + 1)
for k in blkparams.keys():
self.assertEqual(
blkparams[k], blkdata[k],
'For {0}, expected {1}, got {2}'.format(
k, blkparams[k], blkdata[k]))
lstele0 = _np.array(
[1796, 2817, 2818, 3073, 3264, 2754, 2049, 2819, 2820, 3538],
dtype=_np.int32)
tblval0 = _np.array(
[10000, -1, -1, -1, -1, 405, -1, -1, -1, 1029000],
dtype=_np.int32)
self.assertFalse(_np.any(lstele0 - blkdata['cmcids'] != 0))
self.assertEqual(
(blkparams['nele'], blkparams['nval'], blkparams['nt']),
blkdata['tblval'].shape)
self.assertFalse(
_np.any(
tblval0 - blkdata['tblval'][0:blkdata['nele'], 0, 0] != 0))
rmn.burp_close(funit)
def testmrbcvtdecodeKnownValues(self):
"""mrbprm should give known result with known input"""
for mypath, itype, iunit in self.knownValues:
rmn.mrfopt(rmn.FSTOP_MSGLVL, rmn.FSTOPS_MSG_FATAL)
funit = rmn.burp_open(self.getFN(mypath))
nbrp = rmn.mrfnbr(funit)
maxlen = max(64, rmn.mrfmxl(funit))+10
buf = None
handle = 0
handle = rmn.mrfloc(funit, handle)
buf = rmn.mrfget(handle, buf, funit)
params = rmn.mrbhdr(buf)
for iblk in xrange(params['nblk']):
blkparams = rmn.mrbprm(buf, iblk+1)
blkdata = rmn.mrbxtr(buf, iblk+1)
rval = rmn.mrbcvt_decode(blkdata['lstele'],
blkdata['tblval'],
blkparams['datyp'])
rval = rmn.mrbcvt_decode(blkdata,
datyp=blkparams['datyp'])
#TODO: check results
rmn.burp_close(funit)
def testmrbdclKnownValues(self):
"""mrbprm should give known result with known input"""
for mypath, itype, iunit in self.knownValues:
rmn.mrfopt(rmn.FSTOP_MSGLVL, rmn.FSTOPS_MSG_FATAL)
funit = rmn.burp_open(self.getFN(mypath))
nbrp = rmn.mrfnbr(funit)
maxlen = max(64, rmn.mrfmxl(funit))+10
buf = None
handle = 0
handle = rmn.mrfloc(funit, handle)
buf = rmn.mrfget(handle, buf, funit)
params = rmn.mrbhdr(buf)
for iblk in xrange(params['nblk']):
blkparams = rmn.mrbprm(buf, iblk+1)
blkdata = rmn.mrbxtr(buf, iblk+1)
lstelebufr = rmn.mrbdcl(blkdata['lstele'])
lstelebufr0 = _np.array([7004, 11001, 11002, 12001, 12192, 10194,
8001, 11003, 11004, 13210],
dtype=_np.int32)
self.assertFalse(_np.any(lstelebufr0 - lstelebufr != 0))
rmn.burp_close(funit)
def testmrbprmKnownValues(self):
"""mrbprm should give known result with known input"""
for mypath, itype, iunit in self.knownValues:
rmn.mrfopt(rmn.FSTOP_MSGLVL, rmn.FSTOPS_MSG_FATAL)
funit = rmn.burp_open(self.getFN(mypath))
nbrp = rmn.mrfnbr(funit)
maxlen = max(64, rmn.mrfmxl(funit))+10
buf = None
handle = 0
handle = rmn.mrfloc(funit, handle)
buf = rmn.mrfget(handle, buf, funit)
params = rmn.mrbhdr(buf)
for iblk in xrange(params['nblk']):
blkparams = rmn.mrbprm(buf, iblk+1)
blkparams0 = {'nele': 10, 'nbit': 20, 'datyp': 2, 'nval': 17,
'bdesc': 0, 'btyp': 9326, 'bfam': 10, 'nt': 1,
'bit0': 288}
for k in blkparams.keys():
self.assertEqual(blkparams0[k], blkparams[k],
'For {0}, expected {1}, got {2}'
.format(k, blkparams0[k], blkparams[k]))
rmn.burp_close(funit)
def testmrbxtrKnownValues(self):
"""mrbprm should give known result with known input"""
for mypath, itype, iunit in self.knownValues:
rmn.mrfopt(rmn.FSTOP_MSGLVL, rmn.FSTOPS_MSG_FATAL)
funit = rmn.burp_open(self.getFN(mypath))
nbrp = rmn.mrfnbr(funit)
maxlen = max(64, rmn.mrfmxl(funit))+10
buf = None
handle = 0
handle = rmn.mrfloc(funit, handle)
buf = rmn.mrfget(handle, buf, funit)
params = rmn.mrbhdr(buf)
## blkdata = {
## 'lstele' : None,
## 'tblval' : None
## }
for iblk in xrange(params['nblk']):
blkparams = rmn.mrbprm(buf, iblk+1)
## blkdata = rmn.mrbxtr(buf, iblk+1, blkdata['lstele'], blkdata['tblval'])
blkdata = rmn.mrbxtr(buf, iblk+1)
for k in blkparams.keys():
self.assertEqual(blkparams[k], blkdata[k],
'For {0}, expected {1}, got {2}'
.format(k, blkparams[k], blkdata[k]))
lstele0 = _np.array([1796, 2817, 2818, 3073, 3264, 2754, 2049,
2819, 2820, 3538], dtype=_np.int32)
tblval0 = _np.array([10000, -1, -1, -1, -1, 405, -1, -1, -1,
1029000], dtype=_np.int32)
self.assertFalse(_np.any(lstele0 - blkdata['lstele'] != 0))
self.assertEqual((blkparams['nele'], blkparams['nval'],
blkparams['nt']), blkdata['tblval'].shape)
self.assertFalse(_np.any(tblval0 -
blkdata['tblval'][0:blkdata['nele'],0,0]
!= 0))
rmn.burp_close(funit)
