fix tests

Author: jschueller

tests/test_fmi2.py

@@ -1135,12 +1135,15 @@ def test_consecutive_simulation_with_initialize_false():
     enter_continuous_time_mode before resuming integration. Regression
     test to ensure this does not crash or corrupt the FMU state.
     """
-    fmu = load_fmu(REFERENCE_FMU_FMI2_PATH / "Feedthrough.fmu")
+    fmu = load_fmu(REFERENCE_FMU_FMI2_PATH / "BouncingBall.fmu")
 
-    fmu.set('Int32_input', 42)
-    fmu.simulate(0, 0.1, options={"ncp": 1})
-    assert fmu.get('Int32_output')[0] == 42
+    # Sanity check: model has state events
+    nstates, nevents = fmu.get_ode_sizes()
+    assert nevents == 1
+
+    res = fmu.simulate(0, 0.5, options={"ncp": 1})
+    assert fmu.get('h')[0] == res['h'][-1]
 
     # Continue without re-initializing — exercises the fix
-    fmu.simulate(0.1, 0.2, options={"ncp": 1, "initialize": False})
-    assert fmu.get('Int32_output')[0] == 42
+    fmu.simulate(0.5, 1.0, options={"ncp": 1, "initialize": False})
+    assert fmu.get('h')[0] > 0.0

tests/test_fmi3_sim.py

@@ -361,12 +361,15 @@ def test_consecutive_simulation_with_initialize_false(self):
         enter_continuous_time_mode before resuming integration. Regression
         test to ensure this does not crash or corrupt the FMU state.
         """
-        fmu = load_fmu(FMI3_REF_FMU_PATH / "Feedthrough.fmu")
+        fmu = load_fmu(FMI3_REF_FMU_PATH / "BouncingBall.fmu")
+
+        # Sanity check: model has state events
+        nstates, nevents = fmu.get_ode_sizes()
+        assert nevents > 0
 
-        fmu.set('Int32_input', 42)
-        fmu.simulate(0, 0.1, options={"ncp": 1})
-        assert fmu.get('Int32_output')[0] == 42
+        res = fmu.simulate(0, 0.5, options={"ncp": 1})
+        assert fmu.get('h')[0] == res['h'][-1]
 
         # Continue without re-initializing — exercises the fix
-        fmu.simulate(0.1, 0.2, options={"ncp": 1, "initialize": False})
-        assert fmu.get('Int32_output')[0] == 42
+        fmu.simulate(0.5, 1.0, options={"ncp": 1, "initialize": False})
+        assert fmu.get('h')[0] > 0.0