Fix bug when getting a repr of a timestep which is not attached to a …

…session Add test
pynbody · Oct 28, 2023 · 6a5423d · 6a5423d
1 parent fc9e5a8
commit 6a5423d
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/tangos/core/timestep.py b/tangos/core/timestep.py
@@ -76,8 +76,12 @@ def __getitem__(self, halo_identifier):
 
     @property
     def path(self):
-        with Session.object_session(self).no_autoflush:
+        sess = Session.object_session(self)
+        if sess is None:
             return self.simulation.path+"/"+self.extension
+        else:
+            with sess.no_autoflush:
+                return self.simulation.path+"/"+self.extension
 
     @property
     def redshift_cascade(self):

diff --git a/tests/test_timestep.py b/tests/test_timestep.py
@@ -0,0 +1,9 @@
+import tangos
+from tangos import testing
+
+def setup_module():
+    testing.init_blank_db_for_testing()
+def test_timestep_formatter():
+    sim = tangos.core.Simulation('test_sim')
+    ts = tangos.core.TimeStep(sim, "test_timestep")
+    assert ts.path=="test_sim/test_timestep"