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
|
from __future__ import annotations
from xarray import DataArray, DataTree, tutorial
from xarray.testing import assert_identical
from xarray.tests import network
import pytest
@network
class TestLoadDataset:
@pytest.mark.xfail
def test_download_from_github(self, tmp_path) -> None:
cache_dir = tmp_path / tutorial._default_cache_dir_name
ds = tutorial.open_dataset("tiny", cache_dir=cache_dir).load()
tiny = DataArray(range(5), name="tiny").to_dataset()
assert_identical(ds, tiny)
@pytest.mark.xfail
def test_download_from_github_load_without_cache(
self, tmp_path, monkeypatch
) -> None:
cache_dir = tmp_path / tutorial._default_cache_dir_name
ds_nocache = tutorial.open_dataset(
"tiny", cache=False, cache_dir=cache_dir
).load()
ds_cache = tutorial.open_dataset("tiny", cache_dir=cache_dir).load()
assert_identical(ds_cache, ds_nocache)
@network
class TestLoadDataTree:
@pytest.mark.skip("Skipping due to download needed")
def test_download_from_github(self, tmp_path) -> None:
cache_dir = tmp_path / tutorial._default_cache_dir_name
ds = tutorial.open_datatree("tiny", cache_dir=cache_dir).load()
tiny = DataTree.from_dict({"/": DataArray(range(5), name="tiny").to_dataset()})
assert_identical(ds, tiny)
@pytest.mark.skip("Skipping due to download needed")
def test_download_from_github_load_without_cache(
self, tmp_path, monkeypatch
) -> None:
cache_dir = tmp_path / tutorial._default_cache_dir_name
ds_nocache = tutorial.open_datatree(
"tiny", cache=False, cache_dir=cache_dir
).load()
ds_cache = tutorial.open_datatree("tiny", cache_dir=cache_dir).load()
assert_identical(ds_cache, ds_nocache)
|