def __init__(self, dss_solver, objects_by_name, objects_by_class, options, system_paths):
if options["Logging"]["Pre-configured logging"]:
LoggerTag = __name__
else:
LoggerTag = getLoggerTag(options)
self.itr = 0
self.c_seconds = 0
self.c_seconds_old = -1
self._logger = logging.getLogger(LoggerTag)
self._options = options
self._co_convergance_error_tolerance = options['Helics']['Error tolerance']
self._co_convergance_max_iterations = options['Helics']['Max co-iterations']
self._publications = {}
self._subscriptions = {}
self._system_paths = system_paths
self._objects_by_element = objects_by_name
self._objects_by_class = objects_by_class
self._create_helics_federate()
self._dss_solver = dss_solver
self._registerFederateSubscriptions()
self._registerFederatePublications()
helics.helicsFederateEnterExecutingModeIterative(
self._PyDSSfederate,
helics.helics_iteration_request_iterate_if_needed
)
self._logger.info('Entered HELICS execution mode')
def test_iteration_execution_iteration_test():
broker = createBroker(1)
vFed1, fedinfo = createValueFederate(1, "fed0")
# register the publications
pubid = h.helicsFederateRegisterGlobalPublication(
vFed1, "pub1", h.HELICS_DATA_TYPE_DOUBLE, "")
subid = h.helicsFederateRegisterSubscription(vFed1, "pub1", "")
h.helicsFederateSetTimeProperty(vFed1, h.HELICS_PROPERTY_TIME_DELTA, 1.0)
h.helicsFederateEnterInitializingMode(vFed1)
h.helicsPublicationPublishDouble(pubid, 27.0)
comp = h.helicsFederateEnterExecutingModeIterative(
vFed1, h.HELICS_ITERATION_REQUEST_ITERATE_IF_NEEDED)
assert comp == h.HELICS_ITERATION_RESULT_ITERATING
val = h.helicsInputGetDouble(subid)
assert val == 27.0
comp = h.helicsFederateEnterExecutingModeIterative(
vFed1, h.HELICS_ITERATION_REQUEST_ITERATE_IF_NEEDED)
assert comp == h.HELICS_ITERATION_RESULT_NEXT_STEP
val2 = h.helicsInputGetDouble(subid)
assert val2 == val
h.helicsFederateFinalize(vFed1)
destroyFederate(vFed1, fedinfo)
destroyBroker(broker)
def registerPubSubTags(self, pubs=None, subs=None):
self._registerFederateSubscriptions(subs)
self._registerFederatePublications(pubs)
helics.helicsFederateEnterExecutingModeIterative(
self._PyDSSfederate,
helics.helics_iteration_request_iterate_if_needed)
self._logger.info('Entered HELICS execution mode')
def __init__(self, settings, cymepy, Solver, logger):
self.__Logger = logger
self.cympy = cymepy
self.settings = settings
self.Solver = Solver
lf = cymepy.sim.LoadFlow()
lf.Run()
self.itr = 0
self.c_seconds = 0
self.c_seconds_old = -1
self.validTypes = []
for k, v in self.cympy.enums.DeviceType.__dict__.items():
if isinstance(v, int):
self.validTypes.append(k)
self.create_helics_federate()
self.registerPublications()
self.registerSubscriptions()
h.helicsFederateEnterExecutingModeIterative(
self.cymeFederate, h.helics_iteration_request_iterate_if_needed)
logger.info('Entered HELICS execution mode')
def test_bad_inputs_init_error_5():
broker = createBroker(1)
vFed1, fedinfo = createValueFederate(1, "fed0")
# register the publications
h.helicsFederateInfoSetSeparator(fedinfo, "-")
h.helicsFederateSetSeparator(vFed1, "-")
h.helicsFederateRegisterGlobalTypePublication(vFed1, "pub1", "custom1", "")
subid = h.helicsFederateRegisterTypeInput(vFed1, "inp1", "custom2", "")
h.helicsInputAddTarget(subid, "pub1")
h.helicsFederateSetTimeProperty(vFed1, h.HELICS_PROPERTY_TIME_PERIOD, 1.0)
# @test_throws h.HELICSErrorConnectionFailure
with pt.raises(h.HelicsException):
resIt = h.helicsFederateEnterExecutingModeIterative(
vFed1, h.HELICS_ITERATION_REQUEST_NO_ITERATION)
# @test_throws h.HELICSErrorInvalidFunctionCall
with pt.raises(h.HelicsException):
h.helicsFederateRequestTimeIterativeAsync(
vFed1, 1.0, h.HELICS_ITERATION_REQUEST_NO_ITERATION)
# @test_throws h.HELICSErrorInvalidFunctionCall
with pt.raises(h.HelicsException):
res = h.helicsFederateRequestTimeIterativeComplete(vFed1)
h.helicsFederateFinalize(vFed1)
destroyFederate(vFed1, fedinfo)
destroyBroker(broker)
def test_bad_inputs_input_tests():
broker = createBroker(1)
vFed1, fedinfo = createValueFederate(1, "fed0")
# register the publications
pubid = h.helicsFederateRegisterGlobalPublication(
vFed1, "pub1", h.HELICS_DATA_TYPE_DOUBLE, "")
subid = h.helicsFederateRegisterInput(vFed1, "inp1",
h.HELICS_DATA_TYPE_DOUBLE, "")
# @test_throws h.HELICSErrorRegistrationFailure
with pt.raises(h.HelicsException):
subid2 = h.helicsFederateRegisterInput(vFed1, "inp1",
h.HELICS_DATA_TYPE_DOUBLE, "")
h.helicsInputAddTarget(subid, "pub1")
vf2 = h.helicsFederateClone(vFed1)
assert h.helicsFederateGetName(vFed1) == h.helicsFederateGetName(vf2)
h.helicsFederateSetTimeProperty(vFed1, h.HELICS_PROPERTY_TIME_PERIOD, 1.0)
# @test_throws h.HELICSErrorInvalidObject
with pt.raises(h.HelicsException):
ept = h.helicsFederateRegisterEndpoint(vFed1, "ept1", "")
h.helicsFederateEnterInitializingMode(vFed1)
h.helicsPublicationPublishDouble(pubid, 27.0)
comp = h.helicsFederateEnterExecutingModeIterative(
vFed1, h.HELICS_ITERATION_REQUEST_FORCE_ITERATION)
assert comp == h.HELICS_ITERATION_RESULT_ITERATING
val = h.helicsInputGetDouble(subid)
assert val == 27.0
valt = h.helicsInputGetTime(subid)
assert valt == 27.0
comp = h.helicsFederateEnterExecutingModeIterative(
vFed1, h.HELICS_ITERATION_REQUEST_ITERATE_IF_NEEDED)
assert comp == h.HELICS_ITERATION_RESULT_NEXT_STEP
val2 = h.helicsInputGetDouble(subid)
assert val2 == val
# expect error entering initializing Mode again
# @test_throws h.HELICSErrorInvalidFunctionCall
with pt.raises(h.HelicsException):
h.helicsFederateEnterInitializingMode(vFed1)
# expect error entering initializing Mode again
# @test_throws h.HELICSErrorInvalidFunctionCall
with pt.raises(h.HelicsException):
h.helicsFederateEnterInitializingModeAsync(vFed1)
# expect error entering initializing Mode again
# @test_throws h.HELICSErrorInvalidFunctionCall
with pt.raises(h.HelicsException):
h.helicsFederateEnterInitializingModeComplete(vFed1)
h.helicsFederateFinalize(vFed1)
destroyFederate(vFed1, fedinfo)
destroyBroker(broker)