More test fixes

AnyBody-Research-Group · Aug 12, 2021 · 604158d · 604158d
1 parent 5d6d1ae
commit 604158d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git 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
@@ -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)