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
75
|
import os
from pathlib import Path
from typing import Any
import pytest
from cads_api_client import ApiClient
@pytest.fixture
def api_anon_client(api_root_url: str, api_anon_key: str) -> ApiClient:
return ApiClient(url=api_root_url, key=api_anon_key, maximum_tries=0)
def test_features_url_cds_adaptor_area_selection(
tmp_path: Path,
api_anon_client: ApiClient,
) -> None:
collection_id = "test-adaptor-url"
request: dict[str, Any] = {
"variable": "grid_point_altitude",
"reference_dataset": "cru",
"version": "2.1",
}
result_bigger = api_anon_client.retrieve(
collection_id,
**request,
target=str(tmp_path / "bigger.zip"),
)
result_smaller = api_anon_client.retrieve(
collection_id,
**request,
target=str(tmp_path / "smaller.zip"),
area=[50, 0, 40, 10],
)
assert os.path.getsize(result_bigger) > os.path.getsize(result_smaller)
@pytest.mark.parametrize(
"format,expected_extension",
[
("grib", ".grib"),
("netcdf", ".nc"),
],
)
def test_features_mars_cds_adaptor_format(
api_anon_client: ApiClient,
monkeypatch: pytest.MonkeyPatch,
tmp_path: Path,
format: str,
expected_extension: str,
) -> None:
monkeypatch.chdir(tmp_path)
collection_id = "test-adaptor-mars"
request: dict[str, Any] = {
"product_type": "reanalysis",
"variable": "2m_temperature",
"year": "2016",
"month": "01",
"day": "02",
"time": "00:00",
"target": None,
}
result = api_anon_client.retrieve(
collection_id,
**request,
format=format,
)
_, actual_extension = os.path.splitext(result)
assert actual_extension == expected_extension
assert os.path.getsize(result)
|
