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
|
import pytest
from cads_api_client import ApiClient, catalogue
@pytest.fixture
def cat(api_root_url: str, monkeypatch: pytest.MonkeyPatch) -> catalogue.Catalogue:
monkeypatch.setenv("CADS_API_RC", "")
monkeypatch.delenv("CADS_API_KEY", raising=False)
with pytest.warns(UserWarning, match="The API key is missing"):
client = ApiClient(url=api_root_url, maximum_tries=0)
return client.catalogue_api
def test_collections(cat: catalogue.Catalogue) -> None:
res: catalogue.Collections | None = cat.collections()
assert isinstance(res, catalogue.Collections)
assert "collections" in res.json
assert isinstance(res.json["collections"], list)
assert "links" in res.json
assert isinstance(res.json["links"], list)
collection_ids = res.collection_ids()
while len(collection_ids) != res.json["numberMatched"]:
res = res.next()
assert res is not None
collection_ids.extend(res.collection_ids())
expected_collection_id = "reanalysis-era5-single-levels"
assert expected_collection_id in collection_ids
def test_collections_limit(cat: catalogue.Catalogue) -> None:
collections = cat.collections(params={"limit": 1})
res = collections.next()
if res is not None:
assert res.response.status_code == 200
def test_collection(cat: catalogue.Catalogue) -> None:
collection_id = "test-adaptor-mars"
res = cat.collection(collection_id)
assert isinstance(res, catalogue.Collection)
assert "id" in res.json
assert res.id == collection_id
assert "links" in res.json
assert isinstance(res.json["links"], list)
assert res.begin_datetime.isoformat() == "1959-01-01T00:00:00+00:00"
assert res.end_datetime.isoformat() == "2023-05-09T00:00:00+00:00"
|
