diff --git a/tests/test_abcutils.py b/tests/test_abcutils.py index aa81263..a7ffc5b 100644 --- a/tests/test_abcutils.py +++ b/tests/test_abcutils.py @@ -273,8 +273,7 @@ def test_output_dataframe_with_time(self, init_simple_model): app = AnyPyProcess(silent=True) output: AnyPyProcessOutput = app.start_macro(macro)[0] df = output.to_dataframe() - assert df.index.name == "Main.ArmModelStudy.Output.Abscissa.t" - assert df.shape == (100, 11) + assert df.shape == (100, 12) def test_output_dataframe_without_time(self, init_simple_model): diff --git a/tests/test_tools.py b/tests/test_tools.py index 3ce9857..9516c1e 100644 --- a/tests/test_tools.py +++ b/tests/test_tools.py @@ -97,9 +97,9 @@ def test_AnyPyProcessOutput_to_dataframe(): } anypydata = AnyPyProcessOutput(data) df = anypydata.to_dataframe() - assert df.shape == (6, 13) + assert df.shape == (6, 14) df2 = anypydata.to_dataframe(index_var="speciel_length") - assert df2.shape == (5, 34) + assert df2.shape == (5, 35) df3 = anypydata.to_dataframe(index_var=None) assert df3.shape == (1, 39)