From 680dd28f5a9a14b17b415688832110a3dd9ae05a Mon Sep 17 00:00:00 2001 From: Tanmay9223 Date: Tue, 30 Jun 2026 13:05:01 +0530 Subject: [PATCH] fix(tests): move TestVerifierDiscipline above main block (Addresses PR review feedback by ensuring python file-run execution discovers the test class) --- tests/test_sleep_engine.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/tests/test_sleep_engine.py b/tests/test_sleep_engine.py index 1d12a74..1abc54c 100644 --- a/tests/test_sleep_engine.py +++ b/tests/test_sleep_engine.py @@ -1006,8 +1006,7 @@ class TestClaudeCliBackendBare(unittest.TestCase): self.assertIn("Not logged in", getattr(be, "last_call_error", "")) -if __name__ == "__main__": - unittest.main(verbosity=2) + class MockRewardHackingBackend(MockBackend): def attempt(self, task: TaskRecord, skill: str, memory: str, sample_id: int = 0) -> str: @@ -1047,3 +1046,6 @@ class TestVerifierDiscipline(unittest.TestCase): self.assertFalse(res.accepted) self.assertGreater(len(res.rejected_edits), 0) self.assertIn("placeholder", res.rejected_edits[0].content) + +if __name__ == "__main__": + unittest.main(verbosity=2)