Spaces:
Sleeping
Sleeping
File size: 388 Bytes
4d17caa 2f264ab 4d17caa 2f264ab 4d17caa |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
import pytest
from unittest.mock import Mock
from app.pipeline import LanguageModel
@pytest.fixture
def pipeline():
"""
Fixture to provide a mocked LanguageModel instance.
"""
# Create a mock instance of LanguageModel
lm = Mock(spec=LanguageModel)
# Mock the generate_flashcards method
lm.generate_flashcards.return_value = '{"flashcards": []}'
return lm
|