Spaces:
Sleeping
Sleeping
File size: 378 Bytes
4d17caa 8428312 4d17caa 2f264ab 4d17caa 8428312 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 Pipeline
@pytest.fixture
def pipeline():
"""
Fixture to provide a mocked LanguageModel instance.
"""
# Create a mock instance of LanguageModel
lm = Mock(spec=Pipeline)
# Mock the generate_flashcards method
lm.generate_flashcards.return_value = '{"flashcards": []}'
return lm
|