Set up test fixtures
(self)
| 89 | """Test MARS parallel execution functionality""" |
| 90 | |
| 91 | def setUp(self): |
| 92 | """Set up test fixtures""" |
| 93 | self.system_prompt = "You are a mathematical problem solver." |
| 94 | self.test_query = "What is the value of x if 2x + 5 = 15?" |
| 95 | self.model = "mock-model" |
| 96 | |
| 97 | # Set up logging capture for monitoring MARS behavior |
| 98 | self.log_capture = io.StringIO() |
| 99 | self.log_handler = logging.StreamHandler(self.log_capture) |
| 100 | self.log_handler.setLevel(logging.INFO) |
| 101 | |
| 102 | # Add handler to MARS loggers |
| 103 | mars_logger = logging.getLogger('optillm.mars') |
| 104 | mars_logger.addHandler(self.log_handler) |
| 105 | mars_logger.setLevel(logging.INFO) |
| 106 | |
| 107 | # Store original level to restore later |
| 108 | self.original_level = mars_logger.level |
| 109 | |
| 110 | def tearDown(self): |
| 111 | """Clean up test fixtures""" |
nothing calls this directly
no outgoing calls
no test coverage detected