def test_setRunName(self):
writer = SofiePyTableAccess('simple-test-data.h5', '/test')
writer.setRunName('goat')
self.assertEqual(writer.getRuns('goat'), [])
writer.setRunName('goat1', False)
self.assertEqual(writer.getRuns('goat1'), [])
self.assertEqual(writer.getRunName(), '/')
def test_deleteRun(self):
writer = SofiePyTableAccess('simple-test-data.h5', '/test')
self.assertEqual(writer.getRunName(), '/test')
self.assertEqual(writer.getRuns('/'), ['/RunMeta'])
self.assertFalse(writer.deleteRun('goat'))
self.assertFalse(writer.deleteRun('test'))
self.assertEqual(writer.getRuns('/'), ['/RunMeta'])
self.assertFalse(writer.deleteRun('test', recursive=True))
self.assertEqual(writer.getRuns('/'), ['/RunMeta'])
writer.setRunName('/again', True)
self.assertEqual(writer.getRuns('/'), ['/RunMeta'])
self.assertFalse(writer.deleteRun('/again/', recursive=True))
def listdata(options):
if(options.runname == None):
options.runname = '/'
if(options.infile == None):
print "Specify the in file (--infile filename)"
exit()
extension = os.path.splitext(options.infile)
extension = extension[1]
if extension != '.h5':
print "Can only lsit .h5 files: " + options.infile
exit()
theReader = SofiePyTableAccess(options.infile, runName='/')
runs = theReader.getRuns(options.runname)
runs.sort()
for run in runs:
logging.debug('Setting Run: ' + run);
splitRun = os.path.split(run);
theRun, theNode = splitRun
theReader.setRunName(theRun)
meta = theReader.getRunMeta(options.runname)
if meta:
run, description, timestamp = meta[0]
else:
run, description, timestamp = None, None, None
print "\n\n==============================="
print "==============================="
print "RUN: " + theRun + '/' + theNode + "\n-------------------------------"
if theNode == RunMetaName:
print "Description: " + str(description)
print "Time: " + time.strftime('%D, %H:%M:%S',
time.localtime(timestamp))
elif options.verbose:
theReader.setTable(theNode, '')
x = DUMPLINLINES
for row in theReader:
print str(row).lstrip('(').rstrip(')')
if x == 0:
print '.... ONLY THE FIRST ' + \
str(DUMPLINLINES) + ' ENTRIES ARE PRINTED ....'
break;
x -= 1;
else:
print "\n\n==============================="
print "==============================="
print "RUN: " + options.runname + ' DOES NOT EXIST' + \
"\n-------------------------------"
theReader.close()
def test_SofiePyTableAccess(self):
logging.info('AntPyTableAccess Test')
writer = SofiePyTableAccess('simple-test-data.h5', '/test')
self.assertIsInstance(writer, SofiePyTableAccess, 'Root Does not exist')
writer.close()
writer = SofiePyTableAccess('simple-test-data.h5', '/test')
self.assertIsInstance(writer, SofiePyTableAccess, 'Root Exists')
writer = SofiePyTableAccess('simple-test-data.h5', '/testagain')
self.assertIsInstance(writer, SofiePyTableAccess, 'New Root')
writer = SofiePyTableAccess('simple-test-data.h5', '/testagain/new')
self.assertIsInstance(writer, SofiePyTableAccess, 'New Leaf')
writer = SofiePyTableAccess('simple-test-data.h5', '/testagain/new')
self.assertIsInstance(writer, SofiePyTableAccess, 'Repeat Leaf')
writer.setRunName('test/new2')
writer.setRunName('testagain/new3/sub1')
writer.setRunName('testagain/new3/sub1')
writer.setRunName('testagain/new3/sub2')
writer = AntRawDataAccess('simple-test-data.h5', '/test')
writerData = ('1', '2', '3', '4', '5', '6', '7', '8', '9', 111111.101221)
logging.info('Writing the following data: ' + str(writerData))
writer.write(writerData)
logging.info('Came Back from writing.')
writer.close()
writer = AntRawDataAccess('simple-test-data.h5', '/test/AntRawData')
self.assertIsInstance(writer, AntRawDataAccess)
def test_meta(self):
writer = SofiePyTableAccess('simple-test-data.h5', '/test')
self.assertEqual(writer.getRuns('/'), ['/RunMeta'])
data = writer.getRunMeta('test')
logging.debug('Meta Data for the table:' + str(data))
self.assertEqual(len(data),8)
writer.setRunMeta('ourRun', 'The Description')
data = writer.getRunMeta('ourRun')
self.assertEqual(len(data), 8)
logging.debug('The Data:'+str(data))
self.assertEqual('ourRun', data['runName'])
self.assertEqual('The Description', data['runNotes'])
self.assertTrue(writer.setRunMeta('ourRun', 'The New Description'))
data = writer.getRunMeta('ourRun')
logging.debug(data)
self.assertEqual(len(data), 8)
logging.debug('The Data:'+str(data))
self.assertEqual('ourRun', data['runName'])
self.assertEqual('The New Description', data['runNotes'])
data = writer.getRunMeta('test')
self.assertEqual(len(data), 8)
writer = SofiePyTableAccess('simple-test-data.h5', runName='/testagain',
description='our description again')
data = writer.getRunMeta('/testagain')
self.assertEqual(len(data), 8)
self.assertEqual('testagain', data['runName'])
self.assertEqual('our description again', data['runNotes'])
data = writer.getRunMeta('testagain')
self.assertEqual(len(data), 8)
self.assertEqual('testagain', data['runName'])
self.assertEqual('our description again', data['runNotes'])
self.assertIsNone(writer.setRunName('/'));
self.assertIsNone(writer.setRunName('/'));
self.assertIsNone(writer.setTable('/', ''));
