FAOSTAT_data_primap/tests/unit/test_read_data.py

# from src.faostat_data_primap.helper.definitions import domains_and_releases_to_read
import os

from src.faostat_data_primap.read import read_data


def test_read_data_one_domain(tmp_path):
    domains_and_releases_to_read = (("farm_gate_agriculture_energy", "2024-11-14"),)

    read_data(
        domains_and_releases_to_read=domains_and_releases_to_read, save_path=tmp_path
    )

    release_folder = os.listdir(tmp_path)

    # there should be one directory created
    assert len(release_folder) == 1
    # and it starts with "v" (the date changes with each release)
    assert release_folder[0].startswith("v")

    output_files = os.listdir(tmp_path / release_folder[0])
    # in the folder there should be three files
    assert len(output_files) == 3

    # a .yaml, .csv, and .nc file
    required_extensions = {"nc", "csv", "yaml"}
    file_extensions = {file.split(".")[-1] for file in output_files}
    assert required_extensions == file_extensions