Spaces:
Sleeping
Sleeping
import pytest | |
from unittest.mock import Mock | |
from app.pipeline import Pipeline | |
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 | |