Method setUp

tests/test_batching.py:166–174

Set up MLX test fixtures

(self)

    
    @unittest.skipIf(not MLX_AVAILABLE, "MLX not available")
    def setUp(self):
        """Set up MLX test fixtures"""
        self.model_config = MLXModelConfig(
            model_id=TEST_MODEL_MLX,
            max_new_tokens=100
        )
        # Create a real cache manager instead of mock
        from optillm.inference import CacheManager
        self.cache_manager = CacheManager.get_instance(max_size=1)
        
    @unittest.skipIf(not MLX_AVAILABLE, "MLX not available") 
    def test_mlx_batch_creation(self):