async def test_init_models(self, mock_whisper_load, mock_tts_load, mock_llm_load):
        """
        Test that the init_models method initializes the models correctly.
        """
        pipeline = VoicePipeline()

        # Mock the return values of the model loaders
        mock_llm = mock.AsyncMock()
        mock_tokenizer = mock.AsyncMock()
        mock_llm_load.return_value = (mock_llm, mock_tokenizer)

        mock_tts = mock.AsyncMock()
        mock_tts_load.return_value = mock_tts

        mock_stt = mock.AsyncMock()
        mock_whisper_load.return_value = mock_stt

        await pipeline.init_models()

        mock_llm_load.assert_called_once_with(pipeline.llm_model)
        mock_tts_load.assert_called_once_with(pipeline.tts_model)
        mock_whisper_load.assert_called_once_with(pipeline.stt_model)

        assert pipeline.llm is mock_llm
        assert pipeline.tokenizer is mock_tokenizer
        assert pipeline.tts is mock_tts
        assert pipeline.stt is mock_stt

        Test that the init_models method initializes the models correctly.
        
mlx_audio/sts/tests/test_voice_pipeline.py