From 48a1d1a45806f90bcdde90f256253ab1029f1717 Mon Sep 17 00:00:00 2001 From: gmega Date: Mon, 6 Jan 2025 17:01:04 -0300 Subject: [PATCH] fix: broken assertion --- .../core/experiments/tests/test_static_experiment.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/benchmarks/core/experiments/tests/test_static_experiment.py b/benchmarks/core/experiments/tests/test_static_experiment.py index 39ca98a..ad22216 100644 --- a/benchmarks/core/experiments/tests/test_static_experiment.py +++ b/benchmarks/core/experiments/tests/test_static_experiment.py @@ -44,8 +44,13 @@ class MockNode(Node[MockHandle, str]): 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