tests/test_e2e.py

from tests.candidate import complete_interview
from tests.grader import grade
from concurrent.futures import ThreadPoolExecutor

from typing import List


def complete_and_grade_interview(interview_type: str) -> float:
    """
    Complete an interview and return the overall score.

    :param interview_type: Type of the interview.
    :return: Overall score of the interview.
    """
    file_path, _ = complete_interview(interview_type, "test", model="gpt-3.5-turbo")
    feedback = grade(file_path, model="gpt-4o")
    assert feedback["overall_score"] > 0.4
    return feedback["overall_score"]


def test_complete_interview() -> None:
    """
    Test the complete interview process for various interview types.
    """
    interview_types = ["ml_design", "math", "ml_theory", "system_design", "sql", "coding"]
    scores: List[float] = []

    with ThreadPoolExecutor(max_workers=3) as executor:
        futures = [executor.submit(complete_and_grade_interview, it) for it in interview_types]
        for future in futures:
            score = future.result()
            scores.append(score)

    assert sum(scores) / len(scores) > 0.6