fix bad coordinates in history.retrieve

DamCB · Nov 18, 2021 · 7691273 · 7691273
1 parent c104056
commit 7691273
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 4 deletions.
diff --git a/tests/core/test_history.py b/tests/core/test_history.py
@@ -186,7 +186,12 @@ def test_historyHDF5_retrieve():
 def test_historyHDF5_save_every():
     sheet = Sheet("3", *three_faces_sheet())
 
-    history = HistoryHdf5(sheet, save_every=2, dt=1, hf5file="out.hf5",)
+    history = HistoryHdf5(
+        sheet,
+        save_every=2,
+        dt=1,
+        hf5file="out.hf5",
+    )
 
     for element in sheet.datasets:
         assert sheet.datasets[element].shape[0] == history.datasets[element].shape[0]
@@ -214,7 +219,10 @@ def test_historyHDF5_save_every():
 def test_historyHDF5_itemsize():
     sheet = Sheet("3", *three_faces_sheet())
     sheet.vert_df["segment"] = "apical"
-    history = HistoryHdf5(sheet, hf5file="out.hf5",)
+    history = HistoryHdf5(
+        sheet,
+        hf5file="out.hf5",
+    )
 
     for element in sheet.datasets:
         assert sheet.datasets[element].shape[0] == history.datasets[element].shape[0]
@@ -280,6 +288,13 @@ def test_historyHDF5_from_archive():
         os.remove("test.hf5")
 
 
+def test_retrieve_coords():
+    sheet = Sheet("3", *three_faces_sheet())
+    history = History(sheet)
+    history.record()
+    assert history.retrieve(0).coords == sheet.coords
+
+
 def test_to_and_from_archive():
 
     sheet = Sheet("3", *three_faces_sheet())
@@ -298,7 +313,10 @@ def test_to_and_from_archive():
 
 def test_unsaved_col():
     sheet = Sheet("3", *three_faces_sheet())
-    history = HistoryHdf5(sheet, hf5file="test.hf5",)
+    history = HistoryHdf5(
+        sheet,
+        hf5file="test.hf5",
+    )
     history.record()
     history.record()
     sheet.face_df["new_col"] = 0
@@ -309,7 +327,10 @@ def test_unsaved_col():
 
 def test_change_col_types():
     sheet = Sheet("3", *three_faces_sheet())
-    history = HistoryHdf5(sheet, hf5file="test.hf5",)
+    history = HistoryHdf5(
+        sheet,
+        hf5file="test.hf5",
+    )
     history.record()
     history.record()
     sheet.face_df["z"] = "abc"

diff --git a/tyssue/core/history.py b/tyssue/core/history.py
@@ -463,6 +463,7 @@ def retrieve(self, time):
         sheet = type(self.sheet)(
             f"{self.sheet.identifier}_{time:04.3f}", sheet_datasets, self.sheet.specs
         )
+        sheet.coords = self.sheet.coords
         sheet.edge_df.index.rename("edge", inplace=True)
         return sheet