filemetadata.py

#!/usr/bin/env python3
 
 
 
from ROOT import Belle2
import os
import subprocess
import json
import basf2
from basf2.version import get_version
from b2test_utils import safe_process, clean_working_directory
 
basf2.set_random_seed("something important")
 
testFile = 'b2filemetadata.root'
 
with clean_working_directory():
 
    # Generate a small test file
    main = basf2.Path()
    main.add_module('EventInfoSetter', evtNumList=[9, 1], runList=[1, 15], expList=[7, 7])
    main.add_module('RootOutput', outputFileName=testFile, updateFileCatalog=False)
    safe_process(main)
 
    # Check the file meta data (via DataStore)
    read = basf2.Path()
    read.add_module('RootInput', inputFileName=testFile)
    basf2.process(read)
 
    metadata = Belle2.PyStoreObj('FileMetaData', 1)
 
    # print (metadata.getLfn()) #?
    assert 10 == metadata.getNEvents()
    assert 10 == metadata.getNFullEvents()
 
    assert 7 == metadata.getExperimentLow()
    assert 1 == metadata.getRunLow()
    assert 1 == metadata.getEventLow()
    assert 7 == metadata.getExperimentHigh()
    assert 15 == metadata.getRunHigh()
    assert 1 == metadata.getEventHigh()  # highest evt. no in highest run
    assert metadata.containsEvent(7, 1, 1)
    assert metadata.containsEvent(7, 1, 9)
    assert metadata.containsEvent(7, 15, 1)
    assert metadata.containsEvent(7, 1, 10)  # does not exist, but would be plausible
    assert metadata.containsEvent(7, 3, 6)  # same
    assert not metadata.containsEvent(7, 0, 10)  # does not exist, but would be plausible
    assert not metadata.containsEvent(7, 16, 6)  # same
    assert not metadata.containsEvent(0, 15, 6)
 
    assert 0 == metadata.getNParents()
 
    # print (metadata.getDate())
    # assert socket.gethostname() == metadata.getSite()
    # print (metadata.getUser()) #different env variables, not checked
    assert "something important" == metadata.getRandomSeed()
 
    assert get_version() == metadata.getRelease()
    assert metadata.getSteering().startswith('#!/usr/bin/env python3')
    assert metadata.getSteering().strip().endswith('dummystring')
    assert 10 == metadata.getMcEvents()
 
    os.system('touch Belle2FileCatalog.xml')
    assert 0 == os.system('b2file-metadata-add --lfn /logical/file/name ' + testFile)
 
    assert 0 == os.system('b2file-metadata-show ' + testFile)
 
    # Check JSON output (contains steering file, so we cannot use .out)
    metadata_output = subprocess.check_output(['b2file-metadata-show', '--json', testFile])
    m = json.loads(metadata_output.decode('utf-8'))
    assert 7 == m['experimentLow']
    assert 1 == m['runLow']
    assert 1 == m['eventLow']
    assert 7 == m['experimentHigh']
    assert 15 == m['runHigh']
    assert 1 == m['eventHigh']
    assert 'something important' == m['randomSeed']
    assert 10 == m['nEvents']
    assert isinstance(m['nEvents'], int)
    assert 10 == m['nFullEvents']
    assert isinstance(m['nFullEvents'], int)
    assert '/logical/file/name' == m['LFN']
 
    # steering file is in metadata, so we check for existence of this string:
    # dummystring