def test_network__default():
with proj_network_env():
pyproj.network.set_network_enabled()
transformer = Proj(3857)
assert transformer.is_network_enabled == (
os.environ.get("PROJ_NETWORK") == "ON"
)
def test_network__default(transformer):
with proj_network_env():
if pyproj._datadir._USE_GLOBAL_CONTEXT:
pyproj.set_global_context_network()
trans = transformer()
assert trans.is_network_enabled == (
os.environ.get("PROJ_NETWORK") == "ON")
def test_doctests__network():
"""run the examples in the docstrings using the doctest module
that require the network
"""
with proj_network_env():
pyproj.network.set_network_enabled(active=True)
with pytest.warns(DeprecationWarning):
failure_count, _ = doctest.testmod(pyproj.transformer, verbose=True)
assert failure_count == 0, f"{failure_count} of the doctests failed"
def test_transformer_group__network_enabled():
with proj_network_env():
pyproj.network.set_network_enabled(active=True)
trans_group = TransformerGroup(4326, 2964)
assert len(trans_group.unavailable_operations) == 0
assert len(trans_group.transformers) == 10
assert trans_group.best_available
for transformer in trans_group.transformers:
assert transformer.is_network_enabled is True
for operation in transformer.operations:
for grid in operation.grids:
assert grid.available
def test_transformer_group__download_grids__directory(
get_user_data_dir_mock,
download_mock,
tmp_path,
capsys,
):
with proj_network_env():
if pyproj._datadir._USE_GLOBAL_CONTEXT:
pyproj.set_global_context_network(active=False)
trans_group = TransformerGroup(4326, 2964, network=False)
trans_group.download_grids(directory=tmp_path)
get_user_data_dir_mock.assert_not_called()
captured = capsys.readouterr()
assert captured.out == ""
if grids_available(
"us_noaa_alaska.tif",
"ca_nrc_ntv2_0.tif",
check_network=False,
check_all=True,
):
download_mock.assert_not_called()
elif grids_available("us_noaa_alaska.tif", check_network=False):
download_mock.assert_called_with(
file_url="https://cdn.proj.org/ca_nrc_ntv2_0.tif",
short_name="ca_nrc_ntv2_0.tif",
directory=tmp_path,
verbose=False,
)
elif grids_available("ca_nrc_ntv2_0.tif", check_network=False):
download_mock.assert_called_with(
file_url="https://cdn.proj.org/us_noaa_alaska.tif",
short_name="us_noaa_alaska.tif",
directory=tmp_path,
verbose=False,
)
else:
download_mock.assert_has_calls(
[
call(
file_url="https://cdn.proj.org/us_noaa_alaska.tif",
short_name="us_noaa_alaska.tif",
directory=tmp_path,
verbose=False,
),
call(
file_url="https://cdn.proj.org/ca_nrc_ntv2_0.tif",
short_name="ca_nrc_ntv2_0.tif",
directory=tmp_path,
verbose=False,
),
],
any_order=True,
)
def test_transformer_group__network_enabled():
with proj_network_env():
if pyproj._datadir._USE_GLOBAL_CONTEXT:
pyproj.set_global_context_network(active=True)
trans_group = TransformerGroup(4326, 2964, network=True)
assert len(trans_group.unavailable_operations) == 0
assert len(trans_group.transformers) == 10
assert trans_group.best_available
for transformer in trans_group.transformers:
assert transformer.is_network_enabled is True
for operation in transformer.operations:
for grid in operation.grids:
assert grid.available
def test_transformer_group__network_disabled():
with proj_network_env():
if pyproj._datadir._USE_GLOBAL_CONTEXT:
pyproj.network.set_network_enabled(active=False)
trans_group = TransformerGroup(4326, 2964, network=False)
for transformer in trans_group.transformers:
assert transformer.is_network_enabled is False
if grids_available(
"us_noaa_alaska.tif",
"ca_nrc_ntv2_0.tif",
check_network=False,
check_all=True,
):
assert len(trans_group.unavailable_operations) == 0
assert len(trans_group.transformers) == 10
assert (
trans_group.transformers[0].description
== "Inverse of NAD27 to WGS 84 (85) + Alaska Albers"
)
assert trans_group.best_available
elif grids_available("us_noaa_alaska.tif", check_network=False):
assert len(trans_group.unavailable_operations) == 1
assert (
trans_group.transformers[0].description
== "Inverse of NAD27 to WGS 84 (85) + Alaska Albers"
)
assert len(trans_group.transformers) == 9
assert trans_group.best_available
elif grids_available("ca_nrc_ntv2_0.tif", check_network=False):
assert len(trans_group.unavailable_operations) == 1
assert (
trans_group.transformers[0].description
== "Inverse of NAD27 to WGS 84 (7) + Alaska Albers"
)
assert len(trans_group.transformers) == 9
assert not trans_group.best_available
else:
assert len(trans_group.unavailable_operations) == 2
assert (
trans_group.unavailable_operations[0].name
== "Inverse of NAD27 to WGS 84 (85) + Alaska Albers"
)
assert len(trans_group.transformers) == 8
assert not trans_group.best_available
def test_transformer_group__download_grids(get_user_data_dir_mock, tmp_path, capsys):
get_user_data_dir_mock.return_value = str(tmp_path)
with proj_network_env():
if pyproj._datadir._USE_GLOBAL_CONTEXT:
pyproj.network.set_network_enabled(active=False)
trans_group = TransformerGroup(4326, 2964, network=False)
trans_group.download_grids(verbose=True)
captured = capsys.readouterr()
get_user_data_dir_mock.assert_called_with(True)
paths = sorted(Path(path).name for path in glob(str(tmp_path.joinpath("*"))))
if grids_available(
"us_noaa_alaska.tif",
"ca_nrc_ntv2_0.tif",
check_network=False,
check_all=True,
):
assert paths == []
assert captured.out == ""
elif grids_available("us_noaa_alaska.tif", check_network=False):
assert paths == ["ca_nrc_ntv2_0.tif"]
assert (
captured.out == "Downloading: https://cdn.proj.org/ca_nrc_ntv2_0.tif\n"
)
elif grids_available("ca_nrc_ntv2_0.tif", check_network=False):
assert paths == ["us_noaa_alaska.tif"]
assert captured.out == (
"Downloading: https://cdn.proj.org/us_noaa_alaska.tif\n"
)
else:
assert paths == ["ca_nrc_ntv2_0.tif", "us_noaa_alaska.tif"]
assert captured.out == (
"Downloading: https://cdn.proj.org/us_noaa_alaska.tif\n"
"Downloading: https://cdn.proj.org/ca_nrc_ntv2_0.tif\n"
)
# make sure not downloaded again
with proj_env(), patch(
"pyproj.transformer._download_resource_file"
) as download_mock:
append_data_dir(str(tmp_path))
trans_group = TransformerGroup(4326, 2964, network=False)
trans_group.download_grids()
get_user_data_dir_mock.assert_called_with(True)
download_mock.assert_not_called()
def test_doctests():
"""run the examples in the docstrings using the doctest module"""
with warnings.catch_warnings(), proj_network_env():
if pyproj._datadir._USE_GLOBAL_CONTEXT:
pyproj.set_global_context_network(active=True)
warnings.filterwarnings(
"ignore",
"You will likely lose important projection information when",
UserWarning,
)
failure_count_proj, test_count = doctest.testmod(pyproj.proj, verbose=True)
failure_count_crs, test_count_crs = doctest.testmod(pyproj.crs, verbose=True)
failure_count_geod, test_count_geod = doctest.testmod(pyproj.geod, verbose=True)
with pytest.warns(DeprecationWarning):
failure_count_transform, test_count_transform = doctest.testmod(
pyproj.transformer, verbose=True
)
failure_count = (
failure_count_proj
+ failure_count_crs
+ failure_count_geod
+ failure_count_transform
)
expected_failure_count = 0
try:
import shapely # noqa
except ImportError:
# missing shapely
expected_failure_count = 6
# if the below line fails, doctests have failed
assert (
failure_count == expected_failure_count
), f"{failure_count} of the doctests failed"
def test_network__enable(transformer):
with proj_network_env():
pyproj.network.set_network_enabled(active=True)
trans = transformer()
assert trans.is_network_enabled is True
def test_network__enable(transformer):
with proj_network_env():
if pyproj._datadir._USE_GLOBAL_CONTEXT:
pyproj.set_global_context_network(active=True)
trans = transformer(network=True)
assert trans.is_network_enabled is True
def test_network__enable():
with proj_network_env():
pyproj.network.set_network_enabled(active=True)
transformer = Proj(3857)
assert transformer.is_network_enabled is True
def test_network__disable():
with proj_network_env():
if pyproj._datadir._USE_GLOBAL_CONTEXT:
pyproj.set_global_context_network(active=False)
transformer = Proj(3857, network=False)
assert transformer.is_network_enabled is False
def test_network__disable(transformer):
with proj_network_env():
if pyproj._datadir._USE_GLOBAL_CONTEXT:
pyproj.network.set_network_enabled(active=False)
trans = transformer(network=False)
assert trans.is_network_enabled is False
