Revision 84dc076e9ed19b61139e29bfa7626eeec9e1561f authored by Matthew Kolopanis on 14 May 2024, 15:44:27 UTC, committed by Matthew Kolopanis on 24 June 2024, 18:23:23 UTC
1 parent 8cf4de0
conftest.py
# -*- mode: python; coding: utf-8 -*-
# Copyright (c) 2019 Radio Astronomy Software Group
# Licensed under the 2-clause BSD License
"""Testing environment setup and teardown for pytest."""
import os
import pytest
from astropy.coordinates import EarthLocation
from astropy.time import Time
from astropy.utils import iers
import pyuvdata.tests as uvtest
from pyuvdata import UVCal, UVData
from pyuvdata.data import DATA_PATH
@pytest.fixture(autouse=True, scope="session")
def setup_and_teardown_package():
"""Handle possible IERS issues."""
# Do a calculation that requires a current IERS table. This will trigger
# automatic downloading of the IERS table if needed, including trying the
# mirror site in python 3 (but won't redownload if a current one exists).
# If there's not a current IERS table and it can't be downloaded, turn off
# auto downloading for the tests and turn it back on once all tests are
# completed (done by extending auto_max_age).
# Also, the check_warnings function will ignore IERS-related warnings.
try:
t1 = Time.now()
t1.ut1
except Exception:
iers.conf.auto_max_age = None
# Also ensure that we're downloading the site data from astropy
EarthLocation._get_site_registry(force_download=True)
yield
iers.conf.auto_max_age = 30
@pytest.fixture(scope="session")
def uvcalibrate_init_data_main():
"""Make initial uvcalibrate inputs."""
uvdata = UVData()
uvdata.read(
os.path.join(DATA_PATH, "zen.2458098.45361.HH.uvh5_downselected"),
file_type="uvh5",
use_future_array_shapes=True,
)
uvcal = UVCal()
uvcal.read_calfits(
os.path.join(DATA_PATH, "zen.2458098.45361.HH.omni.calfits_downselected"),
use_future_array_shapes=True,
)
yield uvdata, uvcal
@pytest.fixture(scope="function")
def uvcalibrate_init_data(uvcalibrate_init_data_main):
"""Make function level initial uvcalibrate inputs."""
uvdata_in, uvcal_in = uvcalibrate_init_data_main
uvdata = uvdata_in.copy()
uvcal = uvcal_in.copy()
yield uvdata, uvcal
@pytest.fixture(scope="session")
def uvcalibrate_data_main(uvcalibrate_init_data_main):
"""Make uvcalibrate inputs."""
uvdata_in, uvcal_in = uvcalibrate_init_data_main
uvdata = uvdata_in.copy()
uvcal = uvcal_in.copy()
with uvtest.check_warnings(
UserWarning,
match=[
"telescope_location is not set. Using known values for HERA.",
"antenna_positions, antenna_names, antenna_numbers, Nants_telescope are "
"not set or are being overwritten. Using known values for HERA.",
],
):
uvcal.set_telescope_params(overwrite=True)
with uvtest.check_warnings(
UserWarning,
match=[
"Nants_telescope, antenna_diameters, antenna_names, antenna_numbers, "
"antenna_positions, telescope_location, telescope_name are not set or are "
"being overwritten. Using known values for HERA."
],
):
uvdata.set_telescope_params(overwrite=True)
yield uvdata, uvcal
@pytest.fixture(scope="function")
def uvcalibrate_data(uvcalibrate_data_main):
"""Make function level uvcalibrate inputs."""
uvdata_in, uvcal_in = uvcalibrate_data_main
uvdata = uvdata_in.copy()
uvcal = uvcal_in.copy()
yield uvdata, uvcal
Computing file changes ...