utils.py

"""Utilities for running tests."""

import urllib.request
from pathlib import Path

CACHE_DIR = Path(__file__).resolve().parent / "test_data"


def get_cached_level_00(num_files=2):
    """Get a couple real ASCII files to test against."""
    file_urls = (
        "https://metobs-test.ssec.wisc.edu/pub/cache/aoss/tower/level_00/version_00/2020/01/01/aoss_tower.2020-01-01.ascii",
        "https://metobs-test.ssec.wisc.edu/pub/cache/aoss/tower/level_00/version_00/2020/01/02/aoss_tower.2020-01-02.ascii",
    )
    if num_files > len(file_urls):
        msg = f"Maximum of {len(file_urls)} can be loaded."
        raise ValueError(msg)
    file_urls = file_urls[:num_files]

    for u in file_urls:
        fn = CACHE_DIR / u.rsplit("/", 1)[-1]
        if not fn.is_file():
            urllib.request.urlretrieve(u, fn)
        yield fn