def initializeStateManager(topologyCreationFunction, currentScenario):
"""
Initialize StateManager with a Brocooli connection and an empty value store.
:param topologyCreationFunction: Function for topology creation
:param currentScenario: Used underlaying scenario topology like "Masterthesis" or "Alpha"
"""
initializeLogging(level=logging.INFO,
logLevel=False,
logLocation=False,
logTime=True,
logToFile=True)
global scenario
global topology
global observedValuesStore
scenario = currentScenario
topology = topologyCreationFunction()
observedValuesStore = ValueStore("T_{o}")
initializeBroccoli()
from LoggerUtilities import initializeLogging
from TestTopologies import initiateTopologyAlpha
from TestUtilities import playScenarios
logger = logging.getLogger(__name__)
def runAlphaGlobal():
"""Start Alpha_GlobalKnowledge with all scenario cases."""
playScenarios(caseName="Alpha_GlobalKnowledge", topology=initiateTopologyAlpha())
def runAlphaLocal():
"""Start Alpha_LocalKnowledge with all scenario cases on rtu1."""
playScenarios(caseName="Alpha_LocalKnowledge", topology=initiateTopologyAlpha(), rtusToTest={"rtu1"})
def runAlphaGlobalOnly1():
"""Start Alpha_GlobalKnowledge with scenario case 1."""
playScenarios(caseName="Alpha_GlobalKnowledge", topology=initiateTopologyAlpha(), filterFunction=lambda filename: "Scenario1" in filename)
if __name__ == '__main__':
initializeLogging(level=logging.INFO, logLevel=False, logLocation=False, logTime=False)
start = time.time()
runAlphaGlobal()
runAlphaLocal()
runAlphaGlobalOnly1()
end = time.time()
logger.info("All tests completed in %2.3fs." % (end - start))