From a79925a60c96e02ec8b9788b7b810afb186d1dd9 Mon Sep 17 00:00:00 2001 From: Luca Miguel Date: Fri, 21 Jul 2023 15:27:50 +0200 Subject: [PATCH] pre-commit/test fix --- rocket_twin/systems/control/controller_fmu.py | 2 +- rocket_twin/tests/test_controller_fmu.py | 6 +++--- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/rocket_twin/systems/control/controller_fmu.py b/rocket_twin/systems/control/controller_fmu.py index c38a4c1..6634f7e 100644 --- a/rocket_twin/systems/control/controller_fmu.py +++ b/rocket_twin/systems/control/controller_fmu.py @@ -31,7 +31,7 @@ def setup(self, model_path, model_name): self.add_inward("time_var", 0.0, desc="System time", unit="") self.add_inward("time_int", 0.0, desc="Interval between fueling end and launch", unit="") - self.add_inward("time_lnc", 100000., desc="Launch time", unit='') + self.add_inward("time_lnc", 100000.0, desc="Launch time", unit="") self.add_transient("x", der="1") pulling = { diff --git a/rocket_twin/tests/test_controller_fmu.py b/rocket_twin/tests/test_controller_fmu.py index 636113d..12eb924 100644 --- a/rocket_twin/tests/test_controller_fmu.py +++ b/rocket_twin/tests/test_controller_fmu.py @@ -35,8 +35,8 @@ def test_controller_fmu(self): values = { "g_tank.w_out_max": 1.0, "rocket.tank.w_out_max": 0.5, - "controller.time_int": 3., - "rocket.controller.time_int": 3., + "controller.time_int": 3.0, + "rocket.controller.time_int": 3.0, } driver.set_scenario(init=init, values=values) driver.add_recorder(DataFrameRecorder(includes=["rocket.a", "rocket.dyn.a"]), period=1.0) @@ -46,6 +46,6 @@ def test_controller_fmu(self): data = data.drop(["Section", "Status", "Error code"], axis=1) print(data) - np.testing.assert_allclose(sys.rocket.a, 41.0, atol=10 ** (0)) + np.testing.assert_allclose(sys.rocket.a, 40.0, atol=10 ** (0)) np.testing.assert_allclose(sys.g_tank.weight_p, 5.0, atol=10 ** (0)) np.testing.assert_allclose(sys.rocket.tank.weight_p, 0.0, atol=10 ** (0))