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('/', ''));