def testCombineRB5FromTarball(self):
rb52odim.combineRB5FromTarball(self.RB5_TARBALL_DOPVOL1B, self.NEW_H5_TARBALL_DOPVOL1B)
new_rio = _raveio.open(self.NEW_H5_TARBALL_DOPVOL1B)
ref_rio = _raveio.open(self.REF_H5_TARBALL_DOPVOL1B)
self.assertTrue(new_rio.objectType is _rave.Rave_ObjectType_SCAN)
new_scan, ref_scan = new_rio.object, ref_rio.object
validateTopLevel(self, new_scan, ref_scan)
validateScan(self, new_scan, ref_scan)
os.remove(self.NEW_H5_TARBALL_DOPVOL1B)
def testCombineRB5FromTarballReturnRIO(self):
new_rio = rb52odim.combineRB5FromTarball(self.RB5_TARBALL_DOPVOL1B, None, return_rio=True)
ref_rio = _raveio.open(self.REF_H5_TARBALL_DOPVOL1B)
self.assertTrue(new_rio.objectType is _rave.Rave_ObjectType_SCAN)
new_scan, ref_scan = new_rio.object, ref_rio.object
validateTopLevel(self, new_scan, ref_scan)
validateScan(self, new_scan, ref_scan)
def validateMergedPvol(self, new_pvol, iSCAN, ref_RB5_TARBALL):
new_scan = new_pvol.getScan(iSCAN)
# Aha, scan.time NE scan.starttime; getScan() uses parent pvol for time!
new_scan.time=new_scan.starttime
ref_rio = rb52odim.combineRB5FromTarball(ref_RB5_TARBALL, None, return_rio=True)
self.assertTrue(ref_rio.objectType is _rave.Rave_ObjectType_SCAN)
ref_scan = ref_rio.object
# Note: ref_scan only has 1 param to loop upon
validateTopLevel(self, new_scan, ref_scan)
validateScan(self, new_scan, ref_scan)
def testMergeOdimScans2Pvol(self):
rb52odim.combineRB5FromTarball(self.RB5_TARBALL_DOPVOL1A,
self.NEW_H5_TARBALL_DOPVOL1A)
rb52odim.combineRB5FromTarball(self.RB5_TARBALL_DOPVOL1B,
self.NEW_H5_TARBALL_DOPVOL1B)
rb52odim.combineRB5FromTarball(self.RB5_TARBALL_DOPVOL1C,
self.NEW_H5_TARBALL_DOPVOL1C)
FILELIST_H5_TARBALL=[\
self.NEW_H5_TARBALL_DOPVOL1A,\
self.NEW_H5_TARBALL_DOPVOL1B,\
self.NEW_H5_TARBALL_DOPVOL1C,\
]
rio_arr = [_raveio.open(x) for x in FILELIST_H5_TARBALL]
rb52odim.mergeOdimScans2Pvol(rio_arr,
self.NEW_H5_MERGED_PVOL,
interval=5,
taskname='DOPVOL')
new_rio = _raveio.open(self.NEW_H5_MERGED_PVOL)
self.assertTrue(new_rio.objectType is _rave.Rave_ObjectType_PVOL)
new_pvol = new_rio.object
self.assertEqual(new_pvol.getNumberOfScans(), 3)
validateMergedPvol(self, new_pvol, 0, self.RB5_TARBALL_DOPVOL1A)
validateMergedPvol(self, new_pvol, 1, self.RB5_TARBALL_DOPVOL1B)
validateMergedPvol(self, new_pvol, 2, self.RB5_TARBALL_DOPVOL1C)
os.remove(self.NEW_H5_TARBALL_DOPVOL1A)
os.remove(self.NEW_H5_TARBALL_DOPVOL1B)
os.remove(self.NEW_H5_TARBALL_DOPVOL1C)
os.remove(self.NEW_H5_MERGED_PVOL)