1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
|
import numpy as np
import pytest
requests = pytest.importorskip('requests')
import healpy as hp
from astropy.utils import data
pytest.skip('This test require remote data', allow_module_level=True)
@pytest.fixture
def create_reference_alm():
nside = 64
lmax = 96
alm_size = 4753
np.random.seed(123)
input_alm = np.ones(alm_size, dtype=complex)
m = hp.alm2map(input_alm, nside=nside, lmax=lmax)
return lmax, input_alm, m
def test_astropy_download_file():
data.conf.dataurl = "https://healpy.github.io/healpy-data/"
print(
data.get_pkg_data_filename(
"full_weights/healpix_full_weights_nside_0032.fits", package="healpy"
)
)
import unittest.mock
from urllib.error import URLError
def test_map2alm_pixelweights_download_fail(caplog):
nside = 16
m = np.arange(hp.nside2npix(nside))
with unittest.mock.patch('astropy.utils.data.get_pkg_data_filename') as mock_get_pkg_data_filename:
mock_get_pkg_data_filename.side_effect = URLError('Simulated download error')
alm = hp.map2alm(m, use_pixel_weights=True, lmax=3*nside-1)
mock_get_pkg_data_filename.assert_called_once()
assert "Could not download pixel weights" in caplog.text
assert "Proceeding without pixel weights" in caplog.text
assert isinstance(alm, np.ndarray)
def test_pixelweights_local_datapath_missing():
with pytest.raises(RuntimeError):
hp.map2alm(np.zeros(12), use_pixel_weights=True, datapath="datapath/")
def test_pixelweights_local_datapath(tmp_path, create_reference_alm):
lmax, input_alm, m = create_reference_alm
datapath = tmp_path / "datapath" / "full_weights"
datapath.mkdir(parents=True)
pixel_weights_file = requests.get(
"https://github.com/healpy/healpy-data/"
"blob/master/full_weights/"
"healpix_full_weights_nside_0064.fits?raw=true"
)
with open(datapath / "healpix_full_weights_nside_0064.fits", "wb") as f:
f.write(pixel_weights_file.content)
alm = hp.map2alm(
m, use_pixel_weights=True, datapath=tmp_path / "datapath", lmax=lmax
)
np.testing.assert_allclose(input_alm, alm)
|
