Sobes.tech
Junior — Senior
54

Тестовый набор для функции сравнения суммы CSV‑данных

Условие задачи

Создайте несколько юнит‑тестов для функции check_data, которая читает CSV‑файл, вычисляет сумму всех чисел и сравнивает её с заданным значением.

def check_data(file_name: str, target_sum: int = 10) -> bool | NoReturn:
    file_path = Path.join(settings.BASE_PATH, file_name)
    if not file_path.exists():
        raise NoFileError

    df = pd.read_csv(file_path, sep=",", header=None)
    if df.values.sum() == target_sum:
        return True
    return False

Тесты должны проверять корректность работы функции при наличии файла, отсутствии файла и при разных значениях суммы.