def test_federate_tests_federateGeneratedGlobalError():
brk = h.helicsCreateBroker("inproc", "gbrokerc", "--root")
cr = h.helicsCreateCore("inproc", "gcore", "--broker=gbrokerc")
argv = ["", "--corename=gcore"]
fi = h.helicsCreateFederateInfo()
h.helicsFederateInfoLoadFromArgs(fi, argv)
fed1 = h.helicsCreateValueFederate("fed0", fi)
h.helicsFederateInfoFree(fi)
h.helicsFederateEnterExecutingMode(fed1)
h.helicsFederateRequestTime(fed1, 2.0)
h.helicsFederateGlobalError(fed1, 9827, "user generated global error")
with pt.raises(h.HelicsException):
h.helicsFederateRequestTime(fed1, 3.0)
h.helicsFederateDestroy(fed1)
h.helicsCoreDisconnect(cr)
h.helicsBrokerDisconnect(brk)
h.helicsCloseLibrary()
def createBroker(number=1):
initstring = f"-f {number} --name=mainbroker"
# @test_throws h.HELICSErrorInvalidArgument broker = h.helicsCreateBroker("mq", "", initstring)
broker = h.helicsCreateBroker("zmq", "", initstring)
# assert broker is h.Broker
assert h.helicsBrokerIsConnected(broker) is True
return broker
def test_system_tests_broker_logging():
lfile = "log.txt"
rm(lfile, force=True)
broker = h.helicsCreateBroker("inproc", "blog", "--log_level=trace")
h.helicsBrokerSetLogFile(broker, lfile)
h.helicsBrokerDisconnect(broker)
h.helicsCloseLibrary()
assert isfile(lfile)
rm(lfile, force=True)
def create_broker():
initstring = "--federates=2 --name=mainbroker"
broker = h.helicsCreateBroker("zmq", "", initstring)
isconnected = h.helicsBrokerIsConnected(broker)
if isconnected == 1:
pass
return broker
def create_broker():
initstring = "-f 2"
broker = h.helicsCreateBroker("zmq", "", initstring)
isconnected = h.helicsBrokerIsConnected(broker)
if isconnected == 1:
pass
return broker
def vFed():
initstring = "-f 1 --name=mainbroker"
fedinitstring = "--broker=mainbroker --federates=1"
deltat = 0.01
#TODO: should add an assert here about helicsGetVersion
h.helicsGetVersion()
# Create broker #
broker = h.helicsCreateBroker("zmq", "", initstring)
isconnected = h.helicsBrokerIsConnected(broker)
if isconnected == 1:
pass
# Create Federate Info object that describes the federate properties #
fedinfo = h.helicsCreateFederateInfo()
# Set Federate name #
h.helicsFederateInfoSetCoreName(fedinfo, "TestA Core")
# Set core type from string #
h.helicsFederateInfoSetCoreTypeFromString(fedinfo, "zmq")
# Federate init string #
h.helicsFederateInfoSetCoreInitString(fedinfo, fedinitstring)
# Set the message interval (timedelta) for federate. Note th#
# HELICS minimum message time interval is 1 ns and by default
# it uses a time delta of 1 second. What is provided to the
# setTimedelta routine is a multiplier for the default timedelta.
# Set one second message interval #
h.helicsFederateInfoSetTimeProperty(fedinfo, h.helics_property_time_delta,
deltat)
h.helicsFederateInfoSetIntegerProperty(fedinfo,
h.helics_property_int_log_level, 1)
vFed = h.helicsCreateValueFederate("TestA Federate", fedinfo)
yield vFed
h.helicsFederateFinalize(vFed)
state = h.helicsFederateGetState(vFed)
assert state == 3
while (h.helicsBrokerIsConnected(broker)):
time.sleep(1)
h.helicsFederateInfoFree(fedinfo)
h.helicsFederateFree(vFed)
h.helicsCloseLibrary()
def mFed():
import helics as h
initstring = "1 --name=mainbroker"
fedinitstring = "--broker=mainbroker --federates=1"
deltat = 0.01
helicsversion = h.helicsGetVersion()
# Create broker #
broker = h.helicsCreateBroker("zmq", "", initstring)
isconnected = h.helicsBrokerIsConnected(broker)
if isconnected == 1:
pass
# Create Federate Info object that describes the federate properties #
fedinfo = h.helicsFederateInfoCreate()
# Set Federate name #
status = h.helicsFederateInfoSetFederateName(fedinfo, "TestA Federate")
# Set core type from string #
status = h.helicsFederateInfoSetCoreTypeFromString(fedinfo, "zmq")
# Federate init string #
status = h.helicsFederateInfoSetCoreInitString(fedinfo, fedinitstring)
# Set the message interval (timedelta) for federate. Note th#
# HELICS minimum message time interval is 1 ns and by default
# it uses a time delta of 1 second. What is provided to the
# setTimedelta routine is a multiplier for the default timedelta.
# Set one second message interval #
status = h.helicsFederateInfoSetTimeDelta(fedinfo, deltat)
status = h.helicsFederateInfoSetLoggingLevel(fedinfo, 1)
mFed = h.helicsCreateMessageFederate(fedinfo)
yield mFed
status = h.helicsFederateFinalize(mFed)
status, state = h.helicsFederateGetState(mFed)
assert state == 3
while (h.helicsBrokerIsConnected(broker)):
time.sleep(1)
h.helicsFederateFree(mFed)
h.helicsCloseLibrary()
def test_system_test_federate_global_value():
brk = h.helicsCreateBroker("inproc", "gbrokerc", "--root")
cr = h.helicsCreateCore("inproc", "gcore", "--broker=gbrokerc")
argv = ["", "--corename=gcore"]
fi = h.helicsCreateFederateInfo()
h.helicsFederateInfoLoadFromArgs(fi, argv)
fed = h.helicsCreateValueFederate("fed0", fi)
fi2 = h.helicsFederateInfoClone(fi)
h.helicsFederateInfoFree(fi2)
h.helicsFederateInfoFree(fi)
globalVal = "this is a string constant that functions as a global"
globalVal2 = "this is a second string constant that functions as a global"
h.helicsFederateSetGlobal(fed, "testglobal", globalVal)
q = h.helicsCreateQuery("global", "testglobal")
res = h.helicsQueryExecute(q, fed)
assert res == globalVal
h.helicsFederateSetGlobal(fed, "testglobal2", globalVal2)
h.helicsQueryFree(q)
q = h.helicsCreateQuery("global", "testglobal2")
h.helicsQueryExecuteAsync(q, fed)
while h.helicsQueryIsCompleted(q) is False:
time.sleep(0.20)
res = h.helicsQueryExecuteComplete(q)
assert res == globalVal2
q2 = h.helicsCreateQuery("", "isinit")
h.helicsQueryExecuteAsync(q2, fed)
while h.helicsQueryIsCompleted(q2) is False:
time.sleep(0.20)
res = h.helicsQueryExecuteComplete(q2)
assert str(res).lower() == "false"
h.helicsFederateFinalize(fed)
h.helicsCoreDisconnect(cr)
h.helicsBrokerDisconnect(brk)
h.helicsQueryFree(q)
h.helicsQueryFree(q2)
assert h.helicsBrokerIsConnected(brk) is False
h.helicsBrokerDisconnect(brk)
h.helicsCoreDisconnect(cr)
assert h.helicsBrokerIsConnected(brk) is False
h.helicsCloseLibrary()
def mFed():
initstring = "-f 1 --name=mainbroker"
fedinitstring = "--broker=mainbroker --federates=1"
deltat = 0.01
h.helicsGetVersion()
# Create broker #
broker = h.helicsCreateBroker("zmq", "", initstring)
isconnected = h.helicsBrokerIsConnected(broker)
if isconnected == 1:
pass
# Create Federate Info object that describes the federate properties #
fedinfo = h.helicsCreateFederateInfo()
# Set Federate name #
h.helicsFederateInfoSetCoreName(fedinfo, "CoreA Federate")
# Set core type from string #
h.helicsFederateInfoSetCoreTypeFromString(fedinfo, "zmq")
# Federate init string #
h.helicsFederateInfoSetCoreInitString(fedinfo, fedinitstring)
# Set the message interval (timedelta) for federate. Note th#
# HELICS minimum message time interval is 1 ns and by default
# it uses a time delta of 1 second. What is provided to the
# setTimedelta routine is a multiplier for the default timedelta.
# Set one second message interval #
h.helicsFederateInfoSetTimeProperty(fedinfo, h.HELICS_PROPERTY_TIME_DELTA,
deltat)
h.helicsFederateInfoSetIntegerProperty(fedinfo,
h.HELICS_PROPERTY_INT_LOG_LEVEL, 1)
mFed = h.helicsCreateMessageFederate("TestA Federate", fedinfo)
yield mFed
h.helicsFederateFinalize(mFed)
state = h.helicsFederateGetState(mFed)
assert state == 3
while h.helicsBrokerIsConnected(broker):
time.sleep(1)
h.helicsFederateInfoFree(fedinfo)
h.helicsFederateFree(mFed)
h.helicsCloseLibrary()
def AddBroker(core_type="zmq", number_of_federates=1):
initstring = "-f {} --name=mainbroker".format(number_of_federates)
helicsversion = h.helicsGetVersion()
print("HELICS version = {}".format(helicsversion))
# Create broker #
broker = h.helicsCreateBroker(core_type, "", initstring)
isconnected = h.helicsBrokerIsConnected(broker)
if isconnected == 1:
pass
return broker
def AddBroker(core_type="zmq", number_of_federates=1):
initstring = "{} --name=mainbroker".format(number_of_federates)
deltat = 0.01
helicsversion = h.helicsGetVersion()
# Create broker #
broker = h.helicsCreateBroker(core_type, "", initstring)
isconnected = h.helicsBrokerIsConnected(broker)
if isconnected == 1:
pass
return broker
def test_python_api0():
broker = h.helicsCreateBroker("zmq", "", "-f 1 --name=mainbroker")
fedinfo = h.helicsCreateFederateInfo()
assert "HelicsFederateInfo()" in repr(fedinfo)
fedinfo.core_name = "TestFederate"
fedinfo.core_type = "zmq"
fedinfo.core_init = "-f 1 --broker=mainbroker"
mFed = h.helicsCreateCombinationFederate("TestFederate", fedinfo)
assert (
"""HelicsCombinationFederate(name = "TestFederate", state = HelicsFederateState.STARTUP, current_time = -9223372036.854776, n_publications = 0, n_subscriptions = 0, n_endpoints = 0, n_filters = 0, n_pending_messages = 0)"""
in repr(mFed))
_ = mFed.register_endpoint("ep1")
_ = mFed.register_global_endpoint("ep2")
pub = mFed.register_publication("publication", h.HELICS_DATA_TYPE_STRING,
"custom-units")
assert """HelicsPublication(name = "TestFederate/publication", type = "string", units = "custom-units", info = "")""" in repr(
pub)
sub = mFed.register_subscription("subscription", "custom-units")
assert (
"""HelicsInput(name = "_input_3", units = "custom-units", injection_units = "", publication_type = "", type = "", target = "subscription", info = "")"""
in repr(sub))
assert (
"""{ 'CONNECTION_REQUIRED' = 0, 'CONNECTION_OPTIONAL' = 0, 'SINGLE_CONNECTION_ONLY' = 0, 'MULTIPLE_CONNECTIONS_ALLOWED' = 1, 'BUFFER_DATA' = 0, 'STRICT_TYPE_CHECKING' = 0, 'IGNORE_UNIT_MISMATCH' = 0, 'ONLY_TRANSMIT_ON_CHANGE' = 0, 'ONLY_UPDATE_ON_CHANGE' = 0, 'IGNORE_INTERRUPTS' = 0, 'MULTI_INPUT_HANDLING_METHOD' = 0, 'INPUT_PRIORITY_LOCATION' = -1, 'CLEAR_PRIORITY_LIST' = 1, 'CONNECTIONS' = 0 }"""
in repr(sub.option))
sub.option["CONNECTION_REQUIRED"] = 1
sub.option[h.HELICS_HANDLE_OPTION_CONNECTION_REQUIRED] = 1
assert sub.option["CONNECTION_REQUIRED"] == 1
sub.add_target("")
mFed.property[h.HELICS_PROPERTY_TIME_DELTA] = 1.0
assert mFed.property[h.HELICS_PROPERTY_TIME_DELTA] == 1.0
mFed.property["TIME_DELTA"] = 1.0
assert mFed.property["TIME_DELTA"] == 1.0
with pt.raises(h.HelicsException):
mFed.enter_executing_mode()
h.helicsCloseLibrary()
del mFed
del broker
def helicsBroker():
initstring = "-f 1 --name=mainbroker"
# TODO: should add an assert here about helicsGetVersion
h.helicsGetVersion()
# Create broker #
broker = h.helicsCreateBroker("zmq", "", initstring)
isconnected = h.helicsBrokerIsConnected(broker)
if isconnected == 1:
pass
yield broker
h.helicsBrokerFree(broker)
h.helicsCloseLibrary()
def test_system_broker_global_value():
brk = h.helicsCreateBroker("ipc", "gbrokerc", "--root")
globalVal = "this is a string constant that functions as a global"
globalVal2 = "this is a second string constant that functions as a global"
h.helicsBrokerSetGlobal(brk, "testglobal", globalVal)
q = h.helicsCreateQuery("global", "testglobal")
res = h.helicsQueryBrokerExecute(q, brk)
assert res == globalVal
h.helicsBrokerSetGlobal(brk, "testglobal2", globalVal2)
h.helicsQueryFree(q)
q = h.helicsCreateQuery("global", "testglobal2")
res = h.helicsQueryBrokerExecute(q, brk)
assert res == globalVal2
h.helicsBrokerDisconnect(brk)
h.helicsQueryFree(q)
assert h.helicsBrokerIsConnected(brk) is False
h.helicsBrokerFree(brk)
def create_broker(number_of_federates: int = 2):
brokerinitstring = "-f 2"
print("Creating Broker")
broker = h.helicsCreateBroker("zmq", "", brokerinitstring)
print("Created Broker")
if broker.is_connected():
print("Broker created and connected")
print(broker)
while broker.is_connected():
time.sleep(1)
broker.disconnect()
def test_python_api6():
broker = h.helicsCreateBroker("zmq", "broker", "--federates 1")
fi = h.helicsCreateFederateInfo()
fed = h.helicsCreateCombinationFederate("test1", fi)
fed.enter_initializing_mode()
fed.enter_executing_mode()
fed.core.disconnect()
assert fed.core.wait_for_disconnect()
del fed
broker.disconnect()
assert broker.wait_for_disconnect()
del broker
h.helicsCleanupLibrary()
h.helicsCloseLibrary()
def test_broker_api():
assert h.helicsIsCoreTypeAvailable("zmq") == 1
broker1 = h.helicsCreateBroker("zmq", "broker1",
"--federates 3 --loglevel 1")
broker2 = h.helicsBrokerClone(broker1)
address_string = h.helicsBrokerGetAddress(broker1)
assert "tcp://127.0.0.1:23404" in address_string
assert "broker1" in h.helicsBrokerGetIdentifier(broker1)
err = h.helicsErrorInitialize()
h.helicsErrorClear(err)
assert err.error_code == 0
assert h.ffi.string(err.message).decode() == ""
assert h.helicsBrokerIsValid(broker1) == 1
assert h.helicsBrokerIsConnected(broker1) == 1
h.helicsBrokerDisconnect(broker1)
assert h.helicsBrokerIsConnected(broker1) == 0
h.helicsBrokerDisconnect(broker2)
h.helicsBrokerFree(broker1)
h.helicsBrokerFree(broker2)
h.helicsCloseLibrary()
def test_python_api5():
broker = h.helicsCreateBroker("zmq", "broker", "--federates 1")
fi = h.helicsCreateFederateInfo()
fed = h.helicsCreateCombinationFederate("test1", fi)
with pt.raises(h.HelicsException):
fed.register_interfaces("unknownfile.json")
fed.core.disconnect()
assert fed.core.wait_for_disconnect()
del fed
broker.disconnect()
assert broker.wait_for_disconnect()
del broker
h.helicsCleanupLibrary()
h.helicsCloseLibrary()
def test_logging_api():
fi = h.helicsCreateFederateInfo()
broker = h.helicsCreateBroker("zmq", "broker",
"--federates 1 --loglevel 1")
h.helicsFederateInfoSetCoreInitString(fi, "--federates 1")
h.helicsFederateInfoSetIntegerProperty(fi, h.HELICS_PROPERTY_INT_LOG_LEVEL,
5)
fed = h.helicsCreateValueFederate("test1", fi)
userdata = UserData(5)
handle = h.ffi.new_handle(userdata)
h.helicsFederateSetLoggingCallback(fed, logger, handle)
h.helicsFederateEnterExecutingMode(fed)
h.helicsFederateLogInfoMessage(fed, "test MEXAGE")
h.helicsFederateRequestNextStep(fed)
h.helicsFederateLogInfoMessage(fed, "test MEXAGE")
h.helicsFederateRequestNextStep(fed)
h.helicsFederateLogInfoMessage(fed, "test MEXAGE")
h.helicsFederateRequestNextStep(fed)
h.helicsFederateLogInfoMessage(fed, "test MEXAGE")
h.helicsFederateRequestNextStep(fed)
h.helicsFederateFinalize(fed)
assert userdata.x == 9
h.helicsFederateFree(fed)
h.helicsFederateInfoFree(fi)
h.helicsBrokerDisconnect(broker)
h.helicsBrokerFree(broker)
h.helicsCleanupLibrary()
h.helicsCloseLibrary()
def test_system_test_core_global_value1():
brk = h.helicsCreateBroker("zmq", "gbrokerc", "--root")
cr = h.helicsCreateCore("zmq", "gcore", "--broker=gbrokerc")
globalVal = "this is a string constant that functions as a global"
_ = "this is a second string constant that functions as a global"
h.helicsCoreSetGlobal(cr, "testglobal", globalVal)
# q = h.helicsCreateQuery("global", "testglobal")
# TODO: This hangs on core execute
# res = h.helicsQueryCoreExecute(q, cr)
# assert res == globalVal
# h.helicsQueryFree(q)
# @test_broken False
h.helicsCoreDisconnect(cr)
h.helicsBrokerDisconnect(brk)
assert h.helicsBrokerIsConnected(brk) is False
h.helicsCloseLibrary()
def test_system_test_core_global_value2():
brk = h.helicsCreateBroker("zmq", "gbrokerc", "--root")
cr = h.helicsCreateCore("zmq", "gcore", "--broker=gbrokerc")
connected = h.helicsCoreConnect(cr)
assert connected == True
assert h.helicsCoreIsConnected(cr) == True
globalVal = "this is a string constant that functions as a global"
globalVal2 = "this is a second string constant that functions as a global"
h.helicsCoreSetGlobal(cr, "testglobal", globalVal)
q = h.helicsCreateQuery("global", "testglobal")
res = h.helicsQueryCoreExecute(q, cr)
assert res == globalVal
h.helicsCoreSetGlobal(cr, "testglobal2", globalVal2)
h.helicsQueryFree(q)
q = h.helicsCreateQuery("global", "testglobal2")
res = h.helicsQueryCoreExecute(q, cr)
assert res == globalVal2
h.helicsBrokerDisconnect(brk)
h.helicsCoreDisconnect(cr)
h.helicsQueryFree(q)
assert h.helicsBrokerIsConnected(brk) == False
def test_python_api1():
broker = h.helicsCreateBroker("zmq", "", "-f 1 --name=mainbroker")
fedinfo = h.helicsCreateFederateInfo()
assert "HelicsFederateInfo()" in repr(fedinfo)
fedinfo.core_name = "TestFederate"
fedinfo.core_type = "zmq"
fedinfo.core_init = "-f 1 --broker=mainbroker"
mFed = h.helicsCreateCombinationFederate("TestFederate", fedinfo)
assert (
"""HelicsCombinationFederate(name = "TestFederate", state = HelicsFederateState.STARTUP, current_time = -9223372036.854776, n_publications = 0, n_subscriptions = 0, n_endpoints = 0, n_filters = 0, n_pending_messages = 0)"""
in repr(mFed))
_ = mFed.register_endpoint("ep1")
_ = mFed.register_global_endpoint("ep2")
pub = mFed.register_publication("publication", h.HELICS_DATA_TYPE_STRING,
"custom-units")
assert """HelicsPublication(name = "TestFederate/publication", type = "string", units = "custom-units", info = "")""" in repr(
pub)
sub = mFed.register_subscription("TestFederate/publication",
"custom-units")
assert (
"""HelicsInput(name = "_input_3", units = "custom-units", injection_units = "", publication_type = "", type = "", target = "TestFederate/publication", info = "")"""
in repr(sub))
assert (
"""{ 'CONNECTION_REQUIRED' = 0, 'CONNECTION_OPTIONAL' = 0, 'SINGLE_CONNECTION_ONLY' = 0, 'MULTIPLE_CONNECTIONS_ALLOWED' = 1, 'BUFFER_DATA' = 0, 'STRICT_TYPE_CHECKING' = 0, 'IGNORE_UNIT_MISMATCH' = 0, 'ONLY_TRANSMIT_ON_CHANGE' = 0, 'ONLY_UPDATE_ON_CHANGE' = 0, 'IGNORE_INTERRUPTS' = 0, 'MULTI_INPUT_HANDLING_METHOD' = 0, 'INPUT_PRIORITY_LOCATION' = -1, 'CLEAR_PRIORITY_LIST' = 1, 'CONNECTIONS' = 0 }"""
in repr(sub.option))
sub.option["CONNECTION_REQUIRED"] = 1
assert sub.option["CONNECTION_REQUIRED"] == 1
mFed.property[h.HELICS_PROPERTY_TIME_DELTA] = 1.0
assert mFed.property[h.HELICS_PROPERTY_TIME_DELTA] == 1.0
sub.set_default(b"hello")
assert sub.bytes == b"hello"
sub.set_default("world")
assert sub.string == "world"
sub.set_default(0)
assert sub.integer == 0
sub.set_default(True)
assert sub.boolean is True
sub.set_default(1.1)
assert sub.double == 1.1
sub.set_default(2 + 1.1j)
assert sub.complex == 2 + 1.1j
sub.set_default([1.0, 2.0, 3.0])
assert sub.vector == [1.0, 2.0, 3.0]
sub.info = "hello world"
assert sub.info == "hello world"
assert (
"""{ 'CONNECTION_REQUIRED' = 0, 'CONNECTION_OPTIONAL' = 0, 'SINGLE_CONNECTION_ONLY' = 0, 'MULTIPLE_CONNECTIONS_ALLOWED' = 1, 'BUFFER_DATA' = 0, 'STRICT_TYPE_CHECKING' = 0, 'IGNORE_UNIT_MISMATCH' = 0, 'ONLY_TRANSMIT_ON_CHANGE' = 0, 'ONLY_UPDATE_ON_CHANGE' = 0, 'IGNORE_INTERRUPTS' = 0, 'MULTI_INPUT_HANDLING_METHOD' = 0, 'INPUT_PRIORITY_LOCATION' = 0, 'CLEAR_PRIORITY_LIST' = 0, 'CONNECTIONS' = 0 }"""
in repr(mFed.publications["TestFederate/publication"].option))
mFed.publications["TestFederate/publication"].option[
"CONNECTION_REQUIRED"] = 1
assert mFed.publications["TestFederate/publication"].option[
"CONNECTION_REQUIRED"] == 1
mFed.enter_executing_mode()
data = "random-data"
mFed.endpoints["TestFederate/ep1"].default_destination = "ep2"
mFed.endpoints["TestFederate/ep1"].info = "information"
assert mFed.endpoints["TestFederate/ep1"].default_destination == "ep2"
assert mFed.endpoints["TestFederate/ep1"].info == "information"
mFed.endpoints["TestFederate/ep1"].send_data(data, "ep2", 1.0)
mFed.publications["TestFederate/publication"].publish("first-time")
assert mFed.request_time(2.0) == 1.0
try:
assert mFed.subscriptions[
"TestFederate/publication"].bytes == b"first-time"
except Exception as _:
# TODO: this does not work as expected
with pt.raises(AssertionError):
assert mFed.subscriptions[
"TestFederate/publication"].bytes == b"first-time"
assert mFed.has_message()
assert mFed.endpoints["TestFederate/ep1"].has_message() is False
assert mFed.endpoints["ep2"].has_message()
message = mFed.endpoints["ep2"].get_message()
assert message.message_id == 55
assert message.is_valid() is True
assert message.data == "random-data"
assert message.raw_data == b"random-data"
assert len(message.raw_data) == 11
assert message.original_destination == ""
assert message.original_source == "TestFederate/ep1"
assert message.source == "TestFederate/ep1"
assert message.time == 1.0
assert (
"""HelicsMessage(source = "TestFederate/ep1", destination = "ep2", original_source = "TestFederate/ep1", original_destination = "", time = 1.0, id = 55, message = "random-data")"""
in repr(message))
message.append("-random")
assert message.data == "random-data-random"
message.data = "random-data"
assert message.raw_data == b"random-data"
message.raw_data = b"random-data-random"
assert message.data == "random-data-random"
message.message_id = 100
message.source = "earth"
message.destination = "moon"
message.original_source = "hello-world"
message.original_destination = "goodbye-world"
message.time = 2.0
assert message.message_id == 100
assert message.source == "earth"
assert message.destination == "moon"
assert message.original_source == "hello-world"
assert message.original_destination == "goodbye-world"
assert message.time == 2.0
assert (
"""<{ 1 = False, 2 = False, 3 = False, 4 = False, 5 = False, 6 = False, 7 = False, 8 = False, 9 = False, 10 = False, 11 = False, 12 = False, 13 = False, 14 = False, 15 = False }>"""
in repr(message.flag))
message.flag[1] = True
assert message.flag[1] is True
mFed.publications["TestFederate/publication"].publish(1)
assert mFed.request_next_step() == 2.0
assert mFed.subscriptions["TestFederate/publication"].string == "1"
mFed.endpoints["TestFederate/ep1"].send_data(message, "ep2")
mFed.endpoints["TestFederate/ep1"].send_data(message, "ep2", 1.0)
mFed.publications["TestFederate/publication"].publish(1 + 2j)
assert mFed.request_next_step() == 3.0
try:
assert mFed.subscriptions["TestFederate/publication"].string == "1+2j"
except AssertionError:
assert mFed.subscriptions["TestFederate/publication"].string == "[1,2]"
mFed.publications["TestFederate/publication"].publish([1, 2, 3, 4, 5])
assert mFed.request_next_step() == 4.0
assert mFed.subscriptions["TestFederate/publication"].vector == [
1.0, 2.0, 3.0, 4.0, 5.0
]
mFed.publications["TestFederate/publication"].publish(False)
assert mFed.request_next_step() == 5.0
assert mFed.subscriptions["TestFederate/publication"].boolean is False
m = mFed.create_message()
assert (
"""HelicsMessage(source = "", destination = "", original_source = "", original_destination = "", time = 0.0, id = 0, message = "")"""
in repr(m))
mFed.info = "hello-world"
assert mFed.info == "hello-world"
m = mFed.endpoints["ep2"].create_message()
assert (
"""HelicsMessage(source = "", destination = "", original_source = "", original_destination = "", time = 0.0, id = 0, message = "")"""
in repr(m))
mFed.finalize()
del mFed
del broker
def test_error_value_federate_initialize():
initstring = "-f 1 --name=mainbroker"
with pt.raises(h._helics.HelicsException):
h.helicsCreateBroker("mq", "", initstring)
def broker():
brk = h.helicsCreateBroker("zmq", "gbrokertest", "--root")
yield brk
h.helicsBrokerDisconnect(brk)
assert h.helicsBrokerIsConnected(brk) == False
h.helicsCloseLibrary()
# -*- coding: utf-8 -*-
import time
import helics as h
from math import pi
initstring = "-f 2 --name=mainbroker"
fedinitstring = "--broker=mainbroker --federates=1"
deltat = 0.01
helicsversion = h.helicsGetVersion()
print("PI SENDER: Helics version = {}".format(helicsversion))
# Create broker #
print("Creating Broker")
broker = h.helicsCreateBroker("zmq", "", initstring)
print("Created Broker")
print("Checking if Broker is connected")
isconnected = h.helicsBrokerIsConnected(broker)
print("Checked if Broker is connected")
if isconnected == 1:
print("Broker created and connected")
# Create Federate Info object that describes the federate properties #
fedinfo = h.helicsCreateFederateInfo()
# Set Federate name #
h.helicsFederateInfoSetCoreName(fedinfo, "TestA Federate")
# -*- coding: utf-8 -*-
import time
import helics as h
# Create broker #
broker = h.helicsCreateBroker("zmq", "", "-f 2 --name=BeamBroker")
isConnected = h.helicsBrokerIsConnected(broker)
if isConnected == 1:
print("Broker created and connected")
fedInfo = h.helicsCreateFederateInfo()
fedName = "BeamFederateReader"
h.helicsFederateInfoSetCoreName(fedInfo, fedName)
h.helicsFederateInfoSetCoreTypeFromString(fedInfo, "zmq")
h.helicsFederateInfoSetCoreInitString(fedInfo, "--federates=1")
# deltat is multiplied for default timedelta of 1 second
h.helicsFederateInfoSetTimeProperty(fedInfo, h.helics_property_time_delta, 1.0)
cfed = h.helicsCreateCombinationFederate(fedName, fedInfo)
print("Subscribing...")
subs_chargingPlugIn = h.helicsFederateRegisterSubscription(
cfed, "BeamFederate/chargingPlugIn", "string")
subs_chargingPlugOut = h.helicsFederateRegisterSubscription(
cfed, "BeamFederate/chargingPlugOut", "string")
print("Waiting Execution Mode...")
h.helicsFederateEnterInitializingMode(cfed)
h.helicsFederateEnterExecutingMode(cfed)
# start execution loop #
timebin = 300
currenttime = 0
def test_python_api2():
broker = h.helicsCreateBroker("zmq", "broker",
"--federates 1 --loglevel 1")
assert broker.is_connected()
broker.set_global("hello", "world")
broker.data_link("hello", "world")
broker.add_destination_filter_to_endpoint("hello", "world")
broker.add_source_filter_to_endpoint("hello", "world")
try:
assert broker.query("hello", "world") == "#invalid"
except AssertionError:
assert broker.query("hello", "world") == {
"error": {
"code": 404,
"message": "query not valid"
}
}
fi = h.helicsCreateFederateInfo()
fi.core_init = "--federates 1"
fi.property[h.HELICS_PROPERTY_INT_LOG_LEVEL] = 2
fed = h.helicsCreateCombinationFederate("test1", fi)
assert "HelicsCore" in repr(fed.core)
assert 'address = "tcp://127.0.0.1' in repr(fed.core)
assert fed.core.is_connected()
fed.core.set_ready_to_init()
assert "n_publications = 0" in repr(fed)
assert "n_subscriptions = 0" in repr(fed)
assert "n_endpoints = 0" in repr(fed)
assert "n_filters = 0" in repr(fed)
assert fed.property["DELTA"] == 1e-09
assert fed.property["PERIOD"] == 0.0
assert fed.property["OFFSET"] == 0.0
assert fed.property["RT_LAG"] == 0.0
assert fed.property["RT_LEAD"] == 0.0
assert fed.property["RT_TOLERANCE"] == 0.0
assert fed.property["INPUT_DELAY"] == 0.0
assert fed.property["OUTPUT_DELAY"] == 0.0
assert fed.property["MAX_ITERATIONS"] == 50
assert fed.property["LOG_LEVEL"] == 2
assert fed.property["FILE_LOG_LEVEL"] == 2
assert fed.property["CONSOLE_LOG_LEVEL"] == 2
fed.property[h.HELICS_PROPERTY_INT_LOG_LEVEL] = 5
fed.separator = "_"
assert fed.separator == "_"
fed.separator = "/"
assert fed.separator == "/"
assert fed.property[h.HelicsProperty.TIME_DELTA] == 1e-09
assert fed.property[h.HelicsProperty.TIME_PERIOD] == 0.0
assert fed.property[h.HelicsProperty.TIME_OFFSET] == 0.0
assert fed.property[h.HelicsProperty.TIME_RT_LAG] == 0.0
assert fed.property[h.HelicsProperty.TIME_RT_LEAD] == 0.0
assert fed.property[h.HelicsProperty.TIME_RT_TOLERANCE] == 0.0
assert fed.property[h.HelicsProperty.TIME_INPUT_DELAY] == 0.0
assert fed.property[h.HelicsProperty.TIME_OUTPUT_DELAY] == 0.0
assert fed.property[h.HelicsProperty.INT_MAX_ITERATIONS] == 50
assert fed.property[h.HelicsProperty.INT_LOG_LEVEL] == 5
assert fed.property[h.HelicsProperty.INT_FILE_LOG_LEVEL] == 5
assert fed.property[h.HelicsProperty.INT_CONSOLE_LOG_LEVEL] == 5
assert fed.property[h.HelicsProperty.TIME_DELTA.value] == 1e-09
assert fed.property[h.HelicsProperty.TIME_PERIOD.value] == 0.0
assert fed.property[h.HelicsProperty.TIME_OFFSET.value] == 0.0
assert fed.property[h.HelicsProperty.TIME_RT_LAG.value] == 0.0
assert fed.property[h.HelicsProperty.TIME_RT_LEAD.value] == 0.0
assert fed.property[h.HelicsProperty.TIME_RT_TOLERANCE.value] == 0.0
assert fed.property[h.HelicsProperty.TIME_INPUT_DELAY.value] == 0.0
assert fed.property[h.HelicsProperty.TIME_OUTPUT_DELAY.value] == 0.0
assert fed.property[h.HelicsProperty.INT_MAX_ITERATIONS.value] == 50
assert fed.property[h.HelicsProperty.INT_LOG_LEVEL.value] == 5
assert fed.property[h.HelicsProperty.INT_FILE_LOG_LEVEL.value] == 5
assert fed.property[h.HelicsProperty.INT_CONSOLE_LOG_LEVEL.value] == 5
assert "'TIME_DELTA' = 1e-09" in repr(fed.property)
assert "'TIME_PERIOD' = 0.0" in repr(fed.property)
assert "'TIME_OFFSET' = 0.0" in repr(fed.property)
assert "'TIME_RT_LAG' = 0.0" in repr(fed.property)
assert "'TIME_RT_LEAD' = 0.0" in repr(fed.property)
assert "'TIME_RT_TOLERANCE' = 0.0" in repr(fed.property)
assert "'TIME_INPUT_DELAY' = 0.0" in repr(fed.property)
assert "'TIME_OUTPUT_DELAY' = 0.0" in repr(fed.property)
assert "'INT_MAX_ITERATIONS' = 50" in repr(fed.property)
assert "'INT_LOG_LEVEL' = 5" in repr(fed.property)
assert "'INT_FILE_LOG_LEVEL' = 5" in repr(fed.property)
assert "'INT_CONSOLE_LOG_LEVEL' = 5" in repr(fed.property)
assert fed.flag[h.HELICS_FLAG_OBSERVER] is False
assert fed.flag[h.HELICS_FLAG_UNINTERRUPTIBLE] is False
assert fed.flag[h.HELICS_FLAG_INTERRUPTIBLE] is True
assert fed.flag[h.HELICS_FLAG_SOURCE_ONLY] is False
assert fed.flag[h.HELICS_FLAG_ONLY_TRANSMIT_ON_CHANGE] is False
assert fed.flag[h.HELICS_FLAG_ONLY_UPDATE_ON_CHANGE] is False
assert fed.flag[h.HELICS_FLAG_WAIT_FOR_CURRENT_TIME_UPDATE] is False
assert fed.flag[h.HELICS_FLAG_RESTRICTIVE_TIME_POLICY] is False
assert fed.flag[h.HELICS_FLAG_REALTIME] is False
assert fed.flag[h.HELICS_FLAG_SLOW_RESPONDING] is False
assert fed.flag[h.HELICS_FLAG_IGNORE_TIME_MISMATCH_WARNINGS] is False
assert fed.flag[h.HELICS_FLAG_TERMINATE_ON_ERROR] is False
assert fed.flag["TERMINATE_ON_ERROR"] is False
fed.flag["TERMINATE_ON_ERROR"] = False
assert fed.flag[h.HELICS_FLAG_OBSERVER.value] is False
assert fed.flag[h.HELICS_FLAG_UNINTERRUPTIBLE.value] is False
assert fed.flag[h.HELICS_FLAG_INTERRUPTIBLE.value] is True
assert fed.flag[h.HELICS_FLAG_SOURCE_ONLY.value] is False
assert fed.flag[h.HELICS_FLAG_ONLY_TRANSMIT_ON_CHANGE.value] is False
assert fed.flag[h.HELICS_FLAG_ONLY_UPDATE_ON_CHANGE.value] is False
assert fed.flag[h.HELICS_FLAG_WAIT_FOR_CURRENT_TIME_UPDATE.value] is False
assert fed.flag[h.HELICS_FLAG_RESTRICTIVE_TIME_POLICY.value] is False
assert fed.flag[h.HELICS_FLAG_REALTIME.value] is False
assert fed.flag[h.HELICS_FLAG_SLOW_RESPONDING.value] is False
assert fed.flag[h.HELICS_FLAG_IGNORE_TIME_MISMATCH_WARNINGS.value] is False
assert fed.flag[h.HELICS_FLAG_TERMINATE_ON_ERROR.value] is False
assert "'OBSERVER' = False" in repr(fed.flag)
assert "'UNINTERRUPTIBLE' = False" in repr(fed.flag)
assert "'INTERRUPTIBLE' = True" in repr(fed.flag)
assert "'SOURCE_ONLY' = False" in repr(fed.flag)
assert "'ONLY_TRANSMIT_ON_CHANGE' = False" in repr(fed.flag)
assert "'ONLY_UPDATE_ON_CHANGE' = False" in repr(fed.flag)
assert "'WAIT_FOR_CURRENT_TIME_UPDATE' = False" in repr(fed.flag)
assert "'RESTRICTIVE_TIME_POLICY' = False" in repr(fed.flag)
assert "'REALTIME' = False" in repr(fed.flag)
assert "'SLOW_RESPONDING' = False," in repr(fed.flag)
assert "'IGNORE_TIME_MISMATCH_WARNINGS' = False," in repr(fed.flag)
assert "'TERMINATE_ON_ERROR' = False" in repr(fed.flag)
fed.flag[h.HELICS_FLAG_TERMINATE_ON_ERROR] = True
assert fed.flag[h.HELICS_FLAG_TERMINATE_ON_ERROR] is True
fed.core.register_filter(h.HelicsFilterType.DELAY, "")
fed.core.register_cloning_filter("")
fed.register_filter(h.HelicsFilterType.DELAY, "")
fed.register_cloning_filter("")
fed.register_global_filter(h.HelicsFilterType.DELAY, "")
fed.register_global_cloning_filter("")
fed.core.set_global("hello", "world")
try:
assert fed.core.query("broker", "something") == "#invalid"
except AssertionError:
assert fed.core.query("broker", "something") == {
"error": {
"code": 400,
"message": "unrecognized broker query"
}
}
fed.add_dependency("hello")
fed.core.disconnect()
assert fed.core.wait_for_disconnect()
del fed
broker.disconnect()
assert broker.wait_for_disconnect()
del broker
h.helicsCleanupLibrary()
h.helicsCloseLibrary()
def test_python_api7():
broker = h.helicsCreateBroker("zmq", "broker", "--federates 1")
fi = h.helicsCreateFederateInfo()
fed = h.helicsCreateCombinationFederate("test1", fi)
_ = fed.register_filter(h.HELICS_FILTER_TYPE_DELAY, "core1SourceFilter")
assert fed.get_filter_by_name(
"core1SourceFilter").name == fed.get_filter_by_index(0).name
fed.set_global("hello", "world")
# TODO: is_async_operation_completed
assert fed.is_async_operation_completed() is False
fed.enter_initializing_mode_async()
fed.enter_initializing_mode_complete()
assert fed.is_async_operation_completed() is False
fed.enter_executing_mode_async()
fed.enter_executing_mode_complete()
assert fed.is_async_operation_completed() is False
fed.request_time_advance(2.0)
assert fed.current_time == 2.0
fed.request_time_async(4.0)
assert fed.current_time == 2.0
fed.request_time_complete()
assert fed.current_time == 4.0
try:
assert fed.query("hello", "world") == "#invalid"
except AssertionError:
assert fed.query("hello", "world") == {
"error": {
"code": 404,
"message": "query not valid"
}
}
fed.local_error(0, "local")
fed.global_error(0, "global")
fed.log_message("error", logging.ERROR)
fed.log_message("warn", logging.WARN)
fed.log_message("info", logging.INFO)
fed.log_message("debug", logging.DEBUG)
fed.log_message("summary", h.HELICS_LOG_LEVEL_SUMMARY)
fed.finalize_async()
fed.finalize_complete()
fed.core.disconnect()
assert fed.core.wait_for_disconnect()
del fed
broker.disconnect()
assert broker.wait_for_disconnect()
del broker
h.helicsCleanupLibrary()
h.helicsCloseLibrary()
def BMecho_multiCore(cTypeString, federates):
"""This function performs the multicore test for a specific core
type.
Args:
cTypeString (str) - Specific core type, e.g. inproc
federates (int) - The number of federates to create for the echo
multicore test.
Returns:
(null)
"""
logging.info("starting the multicore test for {}".format(cTypeString))
t = Timer(1, timer)
t.cancel()
if h.helicsIsCoreTypeAvailable(cTypeString) == h.helics_false:
t.start()
feds = [f for f in range(0, federates)]
initString = "--log_level=no_print --federates={}".format(federates)
broker = h.helicsCreateBroker(cTypeString, "brokerf", initString)
wcore = h.helicsCreateCore(cTypeString, "",
"--federates=1 --log_level=no_print")
hub = EchoHub_c()
hub_vFed = hub.create_value_federate(h.helicsCoreGetIdentifier(wcore))
hub.initialize(hub_vFed, federates)
leafs = [EchoLeaf_c() for f in range(0, federates)]
cores = []
i = 0
leaf_vFeds = []
logging.info("preparing the federates")
for f in feds:
core = h.helicsCreateCore(cTypeString, None,
"-f 1 --log_level=no_print")
h.helicsCoreConnect(core)
leaf_vFed = leafs[f].create_value_federate(
h.helicsCoreGetIdentifier(core), i)
leafs[f].initialize(leaf_vFed, i)
leaf_vFeds.append(leaf_vFed)
cores.append(core)
i += 1
threads = []
i = 0
logging.info("creating the threads")
for l, f in zip(leaf_vFeds, feds):
x = Thread(target=leafs[f].run, name=leafs[f], args=(len(feds) + 1, l))
threads.append(x)
x.start()
i += 1
time.sleep(0.1)
hub.make_ready(hub_vFed)
logging.info("executing the echo hub")
t.start()
hub.run(len(feds) + 1, hub_vFed)
t.cancel()
logging.info("joining the threads")
for thrd in threads:
thrd.join()
h.helicsBrokerDisconnect(broker)
h.helicsBrokerFree(broker)
logging.info("clearing the cores")
for cr in cores:
h.helicsCoreFree(cr)
cores.clear()
h.helicsCoreFree(wcore)
h.helicsCleanupLibrary()
logging.info("finished the multicore test for {}".format(cTypeString))
# -*- coding: utf-8 -*-
import os
import helics as h
from math import pi
import time
initstring = "-f 2 --loglevel=7"
broker = h.helicsCreateBroker("zmq", "", initstring)
assert h.helicsBrokerIsConnected(broker) is True
fedinfo = h.helicsCreateFederateInfo()
h.helicsFederateInfoSetCoreType(fedinfo, h.helics_core_type_zmq)
h.helicsFederateInfoSetCoreInitString(fedinfo, "--loglevel=7")
h.helicsFederateInfoSetTimeProperty(fedinfo, h.helics_property_time_delta, 1.0)
fed = h.helicsCreateCombinationFederate("sender", fedinfo,)
topicA = h.helicsFederateRegisterGlobalPublication(
fed, "topicA", h.helics_data_type_double, ""
)
h.helicsFederateEnterExecutingMode(fed)
currenttime = 0
for t in range(5, 10 + 1):
while currenttime < t:
currenttime = h.helicsFederateRequestTime(fed, t)
print(f"Sending value = {pi} at time = {currenttime}")
h.helicsPublicationPublishDouble(topicA, pi)
h.helicsFederateFinalize(fed)
h.helicsFederateFree(fed)
def test_misc_api():
fedInfo1 = h.helicsCreateFederateInfo()
h.helicsFederateInfoSetCoreInitString(fedInfo1, "-f 1")
h.helicsFederateInfoSetCoreName(fedInfo1, "core3")
h.helicsFederateInfoSetCoreType(fedInfo1, 3)
h.helicsFederateInfoSetCoreTypeFromString(fedInfo1, "zmq")
h.helicsFederateInfoSetFlagOption(fedInfo1, 1, True)
h.helicsFederateInfoSetTimeProperty(fedInfo1,
h.HELICS_PROPERTY_TIME_INPUT_DELAY,
1.0)
h.helicsFederateInfoSetIntegerProperty(fedInfo1,
h.HELICS_PROPERTY_INT_LOG_LEVEL, 1)
h.helicsFederateInfoSetIntegerProperty(
fedInfo1, h.HELICS_PROPERTY_INT_MAX_ITERATIONS, 100)
h.helicsFederateInfoSetTimeProperty(fedInfo1,
h.HELICS_PROPERTY_TIME_OUTPUT_DELAY,
1.0)
h.helicsFederateInfoSetTimeProperty(fedInfo1,
h.HELICS_PROPERTY_TIME_PERIOD, 1.0)
h.helicsFederateInfoSetTimeProperty(fedInfo1, h.HELICS_PROPERTY_TIME_DELTA,
1.0)
h.helicsFederateInfoSetTimeProperty(fedInfo1,
h.HELICS_PROPERTY_TIME_OFFSET, 0.1)
h.helicsFederateInfoFree(fedInfo1)
broker3 = h.helicsCreateBroker("zmq", "broker3",
"--federates 1 --loglevel 1")
fedInfo2 = h.helicsCreateFederateInfo()
coreInitString = "--federates 1"
h.helicsFederateInfoSetCoreInitString(fedInfo2, coreInitString)
h.helicsFederateInfoSetCoreTypeFromString(fedInfo2, "zmq")
h.helicsFederateInfoSetIntegerProperty(fedInfo2,
h.HELICS_PROPERTY_INT_LOG_LEVEL, 1)
h.helicsFederateInfoSetTimeProperty(fedInfo2, h.HELICS_PROPERTY_TIME_DELTA,
1.0)
fed1 = h.helicsCreateCombinationFederate("fed1", fedInfo2)
fed2 = h.helicsFederateClone(fed1)
_ = h.helicsGetFederateByName("fed1")
h.helicsFederateSetFlagOption(fed2, 1, False)
h.helicsFederateSetTimeProperty(fed2, h.HELICS_PROPERTY_TIME_INPUT_DELAY,
1.0)
h.helicsFederateSetIntegerProperty(fed1, h.HELICS_PROPERTY_INT_LOG_LEVEL,
1)
h.helicsFederateSetIntegerProperty(fed2,
h.HELICS_PROPERTY_INT_MAX_ITERATIONS,
100)
h.helicsFederateSetTimeProperty(fed2, h.HELICS_PROPERTY_TIME_OUTPUT_DELAY,
1.0)
h.helicsFederateSetTimeProperty(fed2, h.HELICS_PROPERTY_TIME_PERIOD, 0.0)
h.helicsFederateSetTimeProperty(fed2, h.HELICS_PROPERTY_TIME_DELTA, 1.0)
_ = h.helicsFederateRegisterCloningFilter(fed1, "fed1/Ep1")
fed1DestinationFilter = h.helicsFederateRegisterFilter(
fed1, h.HELICS_FILTER_TYPE_DELAY, "fed1DestinationFilter")
h.helicsFilterAddDestinationTarget(fed1DestinationFilter, "Ep2")
ep1 = h.helicsFederateRegisterEndpoint(fed1, "Ep1", "string")
ep2 = h.helicsFederateRegisterGlobalEndpoint(fed1, "Ep2", "string")
pub1 = h.helicsFederateRegisterGlobalPublication(fed1, "pub1",
h.HELICS_DATA_TYPE_DOUBLE,
"")
pub2 = h.helicsFederateRegisterGlobalTypePublication(
fed1, "pub2", "complex", "")
sub1 = h.helicsFederateRegisterSubscription(fed1, "pub1")
sub2 = h.helicsFederateRegisterSubscription(fed1, "pub2")
h.helicsInputAddTarget(sub2, "Ep2")
pub3 = h.helicsFederateRegisterPublication(fed1, "pub3",
h.HELICS_DATA_TYPE_STRING, "")
pub1KeyString = h.helicsPublicationGetKey(pub1)
pub1TypeString = h.helicsPublicationGetType(pub1)
pub1UnitsString = h.helicsPublicationGetUnits(pub1)
sub1KeyString = h.helicsSubscriptionGetKey(sub1)
sub1UnitsString = h.helicsInputGetUnits(sub1)
assert "pub1" == pub1KeyString
assert "double" == pub1TypeString
assert "" == pub1UnitsString
assert "pub1" == sub1KeyString
assert "" == sub1UnitsString
fed1SourceFilter = h.helicsFederateRegisterFilter(
fed1, h.HELICS_FILTER_TYPE_DELAY, "fed1SourceFilter")
h.helicsFilterAddSourceTarget(fed1SourceFilter, "Ep2")
h.helicsFilterAddDestinationTarget(fed1SourceFilter, "fed1/Ep1")
h.helicsFilterRemoveTarget(fed1SourceFilter, "fed1/Ep1")
h.helicsFilterAddSourceTarget(fed1SourceFilter, "Ep2")
h.helicsFilterRemoveTarget(fed1SourceFilter, "Ep2")
fed1SourceFilterNameString = h.helicsFilterGetName(fed1SourceFilter)
assert fed1SourceFilterNameString == "fed1/fed1SourceFilter"
sub3 = h.helicsFederateRegisterSubscription(fed1, "fed1/pub3", "")
pub4 = h.helicsFederateRegisterTypePublication(fed1, "pub4", "int", "")
sub4 = h.helicsFederateRegisterSubscription(fed1, "fed1/pub4", "")
pub5 = h.helicsFederateRegisterGlobalTypePublication(
fed1, "pub5", "boolean", "")
sub5 = h.helicsFederateRegisterSubscription(fed1, "pub5", "")
pub6 = h.helicsFederateRegisterGlobalPublication(fed1, "pub6",
h.HELICS_DATA_TYPE_VECTOR,
"")
sub6 = h.helicsFederateRegisterSubscription(fed1, "pub6", "")
pub7 = h.helicsFederateRegisterGlobalPublication(
fed1, "pub7", h.HELICS_DATA_TYPE_NAMED_POINT, "")
sub7 = h.helicsFederateRegisterSubscription(fed1, "pub7", "")
assert """helics.HelicsPublication(name = "pub1", type = "double", units = "", info = "")""" in repr(
pub1)
assert """helics.HelicsPublication(name = "pub2", type = "complex", units = "", info = "")""" in repr(
pub2)
assert """helics.HelicsPublication(name = "fed1/pub3", type = "string", units = "", info = "")""" in repr(
pub3)
assert """helics.HelicsPublication(name = "fed1/pub4", type = "int", units = "", info = "")""" in repr(
pub4)
assert """helics.HelicsPublication(name = "pub5", type = "boolean", units = "", info = "")""" in repr(
pub5)
assert """helics.HelicsPublication(name = "pub6", type = "double_vector", units = "", info = "")""" in repr(
pub6)
assert """helics.HelicsPublication(name = "pub7", type = "named_point", units = "", info = "")""" in repr(
pub7)
assert (
"""helics.HelicsInput(name = "_input_18", units = "", injection_units = "", publication_type = "", type = "", target = "pub7", info = "")"""
in repr(sub7))
h.helicsInputSetDefaultBoolean(sub5, False)
h.helicsInputSetDefaultComplex(sub2, -9.9 + 2.5j)
h.helicsInputSetDefaultDouble(sub1, 3.4)
h.helicsInputSetDefaultInteger(sub4, 6)
h.helicsInputSetDefaultNamedPoint(sub7, "hollow", 20.0)
h.helicsInputSetDefaultString(sub3, "default")
sub6Default = [3.4, 90.9, 4.5]
h.helicsInputSetDefaultVector(sub6, sub6Default)
h.helicsEndpointSubscribe(ep2, "fed1/pub3")
h.helicsFederateEnterInitializingModeAsync(fed1)
rs = h.helicsFederateIsAsyncOperationCompleted(fed1)
if rs == 0:
time.sleep(0.500)
rs = h.helicsFederateIsAsyncOperationCompleted(fed1)
if rs == 0:
time.sleep(0.500)
rs = h.helicsFederateIsAsyncOperationCompleted(fed1)
if rs == 0:
assert True is False
h.helicsFederateEnterInitializingModeComplete(fed1)
h.helicsFederateEnterExecutingModeAsync(fed1)
h.helicsFederateEnterExecutingModeComplete(fed1)
assert (
"""helics.HelicsInput(name = "_input_18", units = "", injection_units = "", publication_type = "named_point", type = "", target = "pub7", info = "")"""
in repr(sub7))
mesg1 = h.helicsFederateCreateMessage(fed1)
h.helicsMessageSetString(mesg1, "Hello")
h.helicsMessageSetSource(mesg1, "fed1/Ep1")
h.helicsMessageSetOriginalSource(mesg1, "fed1/Ep1")
h.helicsMessageSetDestination(mesg1, "Ep2")
h.helicsMessageSetOriginalDestination(mesg1, "Ep2")
h.helicsEndpointSendMessage(ep1, mesg1)
mesg1 = h.helicsFederateCreateMessage(fed1)
h.helicsMessageSetString(mesg1, "There")
h.helicsMessageSetSource(mesg1, "fed1/Ep1")
h.helicsMessageSetOriginalSource(mesg1, "fed1/Ep1")
h.helicsMessageSetDestination(mesg1, "Ep2")
h.helicsMessageSetOriginalDestination(mesg1, "Ep2")
h.helicsEndpointSendMessage(ep1, mesg1)
h.helicsEndpointSetDefaultDestination(ep2, "fed1/Ep1")
ep1NameString = h.helicsEndpointGetName(ep1)
ep1TypeString = h.helicsEndpointGetType(ep1)
assert ep1NameString == "fed1/Ep1"
assert ep1TypeString == "string"
_ = h.helicsFederateGetCoreObject(fed1)
fed1Time = h.helicsFederateGetCurrentTime(fed1)
assert fed1Time == 0.0
fed1EndpointCount = h.helicsFederateGetEndpointCount(fed1)
assert fed1EndpointCount == 2
fed1NameString = h.helicsFederateGetName(fed1)
assert fed1NameString == "fed1"
fed1State = h.helicsFederateGetState(fed1)
assert fed1State == 2
fed1PubCount = h.helicsFederateGetPublicationCount(fed1)
assert fed1PubCount == 7
fed1SubCount = h.helicsFederateGetInputCount(fed1)
assert fed1SubCount == 7
h.helicsPublicationPublishBoolean(pub5, True)
h.helicsPublicationPublishComplex(pub2, 5.6 + -0.67j)
h.helicsPublicationPublishDouble(pub1, 457.234)
h.helicsPublicationPublishInteger(pub4, 1)
h.helicsPublicationPublishNamedPoint(pub7, "Blah Blah", 20.0)
h.helicsPublicationPublishString(pub3, "Mayhem")
pub6Vector = [4.5, 56.5]
h.helicsPublicationPublishVector(pub6, pub6Vector)
time.sleep(0.500)
h.helicsFederateRequestTimeAsync(fed1, 1.0)
returnTime = h.helicsFederateRequestTimeComplete(fed1)
assert returnTime == 1.0
ep2MsgCount = h.helicsEndpointPendingMessages(ep2)
assert ep2MsgCount == 2
ep2HasMsg = h.helicsEndpointHasMessage(ep2)
assert ep2HasMsg == 1
msg2 = h.helicsEndpointGetMessage(ep2)
assert h.helicsMessageGetTime(msg2) == 1.0
assert h.helicsMessageGetString(msg2) == "Hello"
assert h.helicsMessageGetOriginalSource(msg2) == "fed1/Ep1"
assert h.helicsMessageGetSource(msg2) == "fed1/Ep1"
assert h.helicsMessageGetDestination(msg2) == "Ep2"
assert h.helicsMessageGetOriginalDestination(msg2) == "Ep2"
fed1MsgCount = h.helicsFederatePendingMessages(fed1)
assert fed1MsgCount == 1
assert h.helicsFederateHasMessage(fed1) == 1
msg3 = h.helicsFederateGetMessage(fed1)
assert h.helicsMessageGetTime(msg3) == 1.0
assert h.helicsMessageGetString(msg3) == "There"
assert h.helicsMessageGetOriginalSource(msg3) == "fed1/Ep1"
assert h.helicsMessageGetSource(msg3) == "fed1/Ep1"
assert h.helicsMessageGetDestination(msg3) == "Ep2"
assert h.helicsMessageGetOriginalDestination(msg3) == "Ep2"
sub1Updated = h.helicsInputIsUpdated(sub1)
# TODO: figure out why this test is broken
assert sub1Updated is False
# TODO: figure out why this test is broken
assert h.helicsInputLastUpdateTime(sub2) == 0.0
# assert h.helicsInputGetComplex(sub2) == 5.6 - 0.67j
# assert h.helicsInputGetDouble(sub1) == 457.234
# assert h.helicsInputGetInteger(sub4) == 1
sub7PointString, sub7DoubleValue = h.helicsInputGetNamedPoint(sub7)
# assert sub7PointString == "Blah Blah"
assert sub7DoubleValue == 20.0
# assert h.helicsInputGetBoolean(sub5) == True
# assert h.helicsInputGetString(sub3) == "Mayhem"
sub3ValueSize = h.helicsInputGetRawValueSize(sub3)
# assert sub3ValueSize == 6
# assert h.helicsInputGetVector(sub6) == [4.5, 56.5]
h.helicsFederateFinalize(fed1)
h.helicsFederateFinalize(fed2)
h.helicsFederateFree(fed1)
h.helicsFederateFinalize(fed2)
h.helicsFederateFree(fed2)
h.helicsFederateInfoFree(fedInfo2)
h.helicsBrokerDisconnect(broker3)
h.helicsBrokerFree(broker3)
h.helicsCleanupLibrary()
h.helicsCloseLibrary()
