conftest.py

import pytest

# from glob import glob
import yaml

path_to_data = "test/data/"
# path_to_data = 'data/'


# TODO: this fixture is no longer used, move it's data into sir_records.yml
@pytest.fixture
def slotfiles():
    # slotfiles_files = glob(path_to_data + 'slots_*.yml')
    slotfiles_files = ["slots_austria.yml", "slots_belgium.yml", "slots_netherlands.yml"]
    slotfile_list = []
    for slotfile in slotfiles_files:
        with open(path_to_data + slotfile) as f:
            text = yaml.safe_load(f.read())

        slotfile_list.append(text)

    return slotfile_list


@pytest.fixture
def expanding_slots():
    with open(path_to_data + "expanding_slots.yml") as f:
        slots = yaml.safe_load(f.read())

    return slots


@pytest.fixture
def sir_records():
    with open(path_to_data + "sir_records.yml") as f:
        slots = yaml.safe_load(f.read())

    return slots


@pytest.fixture
def sim_records():
    with open(path_to_data + "sim_records.yml") as f:
        slots = yaml.safe_load(f.read())

    return slots


@pytest.fixture
def aircraft_configuration_strings():
    with open(path_to_data + "aircraft_configuration_strings.yml") as f:
        aircraft_configuration_strings = yaml.safe_load(f.read())

    return aircraft_configuration_strings