tests/test_pipeline.py

import pytest

from evaluation.config import GeneratorConfig, PipelineConfig, RetrieverConfig
from evaluation.pipeline import RAGPipeline


def test_pipeline_init():
    # Using bm25 + dummy index path
    cfg = PipelineConfig(
        retriever=RetrieverConfig(name="bm25", bm25_idx="dummy"),
        generator=GeneratorConfig(model_name="google/flan-t5-base"),
    )
    pipeline = RAGPipeline(cfg)
    assert pipeline.retriever is not None
    assert pipeline.generator is not None