Method setUp

tests/test_batching.py:34–58

Set up test fixtures

(self)

    """Test the core RequestBatcher functionality"""
    
    def setUp(self):
        """Set up test fixtures"""
        self.batcher = RequestBatcher(max_batch_size=4, max_wait_ms=100)
        self.test_responses = []
        
        def mock_processor(requests):
            """Mock batch processor that returns simple responses"""
            responses = []
            for i, req in enumerate(requests):
                responses.append({
                    "id": f"test-{i}",
                    "object": "chat.completion",
                    "choices": [{
                        "index": 0,
                        "message": {
                            "role": "assistant",
                            "content": f"Response to request {i}"
                        },
                        "finish_reason": "stop"
                    }],
                    "usage": {"completion_tokens": 10, "total_tokens": 20}
                })
            return responses
        
        self.batcher.set_processor(mock_processor)
    
    def tearDown(self):
        """Clean up after tests"""