diff --git a/README.md b/README.md index f01df22..52292de 100644 --- a/README.md +++ b/README.md @@ -46,7 +46,7 @@ List of all available filters: from gpuhunt import Catalog catalog = Catalog() -catalog.load(version="20230831") +catalog.load(version="20240310") items = catalog.query() print(*items, sep="\n") diff --git a/src/integrity_tests/test_datacrunch.py b/src/integrity_tests/test_datacrunch.py index a471a7a..5fe796a 100644 --- a/src/integrity_tests/test_datacrunch.py +++ b/src/integrity_tests/test_datacrunch.py @@ -20,7 +20,11 @@ def select_row(rows, name: str) -> List[str]: def test_locations(data_rows): - expected = set(("FIN-01", "ICE-01")) + expected = { + "FIN-01", + "FIN-02", + "ICE-01", + } locations = select_row(data_rows, "location") assert set(locations) == expected diff --git a/src/tests/providers/test_datacrunch.py b/src/tests/providers/test_datacrunch.py index a13ca9b..6756bb8 100644 --- a/src/tests/providers/test_datacrunch.py +++ b/src/tests/providers/test_datacrunch.py @@ -147,6 +147,7 @@ def locations(): # datacrunch.locations.get() return [ {"code": "FIN-01", "name": "Finland 1", "country_code": "FI"}, + {"code": "FIN-02", "name": "Finland 2", "country_code": "FI"}, {"code": "ICE-01", "name": "Iceland 1", "country_code": "IS"}, ]