fix: broken assertion

codex-storage · Jan 6, 2025 · 48a1d1a · 48a1d1a
1 parent 59dcba5
commit 48a1d1a
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/benchmarks/core/experiments/tests/test_static_experiment.py b/benchmarks/core/experiments/tests/test_static_experiment.py
@@ -44,8 +44,13 @@ def leech(self, handle: MockHandle):
         return MockDownloadHandle(self)
 
     def remove(self, handle: MockHandle):
-        assert self.seeding is not None
-        assert self.leeching == handle or self.seeding[0] == handle
+        if self.leeching is not None:
+            assert self.leeching == handle
+        elif self.seeding is not None:
+            assert self.seeding[0] == handle
+        else:
+            raise Exception('Either leech or seed must be called before attempting a remove')
+
         self.remove_was_called = True