Skip to content

Commit acfb111

Browse files
committed
Add some very simple tests for variablelist.
1 parent dce21f7 commit acfb111

1 file changed

Lines changed: 33 additions & 0 deletions

File tree

tests/test_variablelist.py

Lines changed: 33 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,33 @@
1+
import pytest
2+
3+
from bris.conventions.metno import Metno
4+
from bris.conventions.variable_list import VariableList
5+
6+
7+
def test_initialization():
8+
"""Test initialization of VariableList."""
9+
10+
vl = VariableList(anemoi_names=[])
11+
assert isinstance(vl.conventions, Metno)
12+
13+
14+
def test_dimensions_property():
15+
"""Test the dimensions property."""
16+
vl = VariableList(anemoi_names=["2t"])
17+
dimensions = vl.dimensions
18+
assert isinstance(dimensions, dict)
19+
assert "height" in dimensions["height"][0]
20+
assert dimensions["height"][1] == [2]
21+
assert vl._ncname_to_level_dim == {"air_temperature_2m": "height"}
22+
23+
24+
def test_get_level_index():
25+
vl = VariableList(anemoi_names=["2t", "10u", "10v"])
26+
level = vl.get_level_index(anemoi_name="2t")
27+
assert level == 0
28+
29+
30+
def test_get_ncname_from_anemoi_name():
31+
vl = VariableList(anemoi_names=["2t"])
32+
ncname = vl.get_ncname_from_anemoi_name("2t")
33+
assert ncname == "air_temperature_2m"

0 commit comments

Comments
 (0)