def test_store_connections(): """Test that it is possible to store connections in a config file""" sim = Sim() wave_sys = SquareWave() wave_sys.inputs.amplitude = 50 wave_sys.inputs.freq = 0.1 sim.add_system(wave_sys,"wave_sys") msd = MassSpringDamper() msd.inputs.b = 80 msd.inputs.m = 50 msd.inputs.f = 0 sim.add_system(msd,"msd") wave_sys.connections.add_connection("signal", msd, "f") # In this test a temp file is used, it should be deleted automatically # after the test. file = tempfile.NamedTemporaryFile() file.close() sim.save_config(file.name) file2 = open(file.name) simdict = json.load(file2) file2.close() print(simdict) cons = simdict["systems"]["wave_sys"]["connections"] assert cons[0] == ["signal", "msd", "f"]
def test_store_connections(): """Test that it is possible to store connections in a config file""" sim = Sim() wave_sys = SquareWave() wave_sys.inputs.amplitude = 50 wave_sys.inputs.freq = 0.1 sim.add_system(wave_sys, "wave_sys") msd = MassSpringDamper() msd.inputs.b = 80 msd.inputs.m = 50 msd.inputs.f = 0 sim.add_system(msd, "msd") wave_sys.connections.add_connection("signal", msd, "f") # In this test a temp file is used, it should be deleted automatically # after the test. file = tempfile.NamedTemporaryFile() file.close() sim.save_config(file.name) file2 = open(file.name) simdict = json.load(file2) file2.close() print(simdict) cons = simdict["systems"]["wave_sys"]["connections"] assert cons[0] == ["signal", "msd", "f"]
def test_store_config_composite_ports(): """Test that it is possible to store the ports of a composite systems to a file. """ sys = CompositeSpring() sim = Sim() sim.add_system(sys, "composite_root") # In this test a temp file is used, it should be deleted automatically # after the test. file = tempfile.NamedTemporaryFile() file.close() sim.save_config(file.name) file2 = open(file.name) simdict = json.load(file2) file2.close() print(simdict) ports = simdict["systems"]["composite_root"]["ports"] forceport = ports["in"]["force"] assert forceport["type"] == "scalar" assert forceport["value"] == 0 assert forceport["description"] == "force acting on mass" assert forceport["connections"][0]["subsystem"] == "msd" assert forceport["connections"][0]["input"] == "f" posport = ports["out"]["position"] assert posport["type"] == "scalar" assert posport["value"] == 0 assert posport["description"] == "Position" assert posport["connections"][0]["subsystem"] == "msd" assert posport["connections"][0]["output"] == "x1"
def test_store_config_composite_ports(): """Test that it is possible to store the ports of a composite systems to a file. """ sys = CompositeSpring() sim = Sim() sim.add_system(sys, "composite_root") # In this test a temp file is used, it should be deleted automatically # after the test. file = tempfile.NamedTemporaryFile() file.close() sim.save_config(file.name) file2 = open(file.name) simdict = json.load(file2) file2.close() print(simdict) ports = simdict["systems"]["composite_root"]["ports"] forceport = ports["in"]["force"] assert forceport["type"] == "scalar" assert forceport["value"] == 0 assert forceport["description"] == "force acting on mass" assert forceport["connections"][0]["subsystem"] == "msd" assert forceport["connections"][0]["input"] == "f" posport = ports["out"]["position"] assert posport["type"] == "scalar" assert posport["value"] == 0 assert posport["description"] == "Position" assert posport["connections"][0]["subsystem"] == "msd" assert posport["connections"][0]["output"] == "x1"
def test_store_config(test_class): """Test that it is possible to store the simulation to a file. """ sys = test_class() sim = Sim() sim.add_system(sys) sys.inputs.a = 1.234 sys.store("x") # In this test a temp file is used, it should be deleted automatically # after the test. file = tempfile.NamedTemporaryFile() file.close() sim.save_config(file.name) file2 = open(file.name) simdict = json.load(file2) file2.close() assert simdict["systems"]["vanderpol"]["inputs"]["a"] == 1.234 assert "x" in simdict["systems"]["vanderpol"]["stores"]
def test_store_config(test_class): """Test that it is possible to store the simulation to a file. """ sys = test_class() sim = Sim() sim.add_system(sys) sys.inputs.a = 1.234 sys.store("x") # In this test a temp file is used, it should be deleted automatically # after the test. file = tempfile.NamedTemporaryFile() file.close() sim.save_config(file.name) file2 = open(file.name) simdict = json.load(file2) file2.close() assert simdict["systems"]["vanderpol"]["inputs"]["a"] == 1.234 assert "x" in simdict["systems"]["vanderpol"]["stores"]
def test_store_connections_composite(): """Test that it is possible to store subsystem connections in a config file. """ sim = Sim() sys = ControlledSpring() sim.add_system(sys, "controlled_spring") # In this test a temp file is used, it should be deleted automatically # after the test. file = tempfile.NamedTemporaryFile() file.close() sim.save_config(file.name) file2 = open(file.name) simdict = json.load(file2) file2.close() print(simdict) cons = simdict["systems"]["controlled_spring"]["subsystems"]["wave_sys"]["connections"] assert cons[0] == ["signal", "msd", "f"]
def test_store_connections_composite(): """Test that it is possible to store subsystem connections in a config file. """ sim = Sim() sys = ControlledSpring() sim.add_system(sys, "controlled_spring") # In this test a temp file is used, it should be deleted automatically # after the test. file = tempfile.NamedTemporaryFile() file.close() sim.save_config(file.name) file2 = open(file.name) simdict = json.load(file2) file2.close() print(simdict) cons = simdict["systems"]["controlled_spring"]["subsystems"]["wave_sys"][ "connections"] assert cons[0] == ["signal", "msd", "f"]