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