fix tests by making other outputs respect reset

joezuntz · Dec 4, 2024 · 695a9fb · 695a9fb
1 parent 52186b5
commit 695a9fb
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/cosmosis/output/astropy_output.py b/cosmosis/output/astropy_output.py
@@ -31,3 +31,6 @@ def _write_final(self, key, value, comment):
     def _write_comment(self, comment):
         self.table.meta["comments"] = self.table.meta.get("comments", []) + [comment]
 
+    def reset_to_chain_start(self):
+         self.table.remove_rows(slice(None))
+
diff --git a/cosmosis/output/in_memory_output.py b/cosmosis/output/in_memory_output.py
@@ -38,5 +38,8 @@ def from_options(cls, options, resume=False):
     @classmethod
     def load_from_options(cls, options):
         raise ValueError("No output was saved from this run")
+
+    def reset_to_chain_start(self):
+        self.rows = []