def test_create_from_expr3():
src = ifs_source.read("var88", "sqrt(sq(var128131)+sq(var128132))")
eq_(src.get_grib_code(), grib_code(88, 128))
eq_(src.get_root_codes(), [grib_code(131, 128), grib_code(132, 128)])
eq_(getattr(src, "expr"), "var88=sqrt(sq(var131)+sq(var132))")
eq_(src.grid(), "spec")
eq_(src.spatial_dims, 3)
def test_create_from_expr():
src = ifs_source.read("88.128", "sqrt(sq(var165)+sq(var166))")
eq_(src.get_grib_code(), grib_code(88, 128))
eq_(src.get_root_codes(), [grib_code(165, 128), grib_code(166, 128)])
eq_(getattr(src, "expr"), "var88=sqrt(sq(var165)+sq(var166))")
eq_(src.grid(), "point")
eq_(src.spatial_dims, 2)
def test_create_from_expr3():
src = ifs_source.read("var88", "sqrt(sq(var128131)+sq(var128132))")
assert src.get_grib_code() == grib_code(88, 128)
assert set(src.get_root_codes()) == {
grib_code(131, 128), grib_code(132, 128)
}
assert getattr(src, "expr") == "var88=sqrt(sq(var131)+sq(var132))"
assert src.grid() == "spec"
assert src.spatial_dims == 3
def test_create_from_expr():
src = ifs_source.read("88.128", "sqrt(sq(var165)+sq(var166))")
assert src.get_grib_code() == grib_code(88, 128)
assert set(src.get_root_codes()) == {
grib_code(165, 128), grib_code(166, 128)
}
assert getattr(src, "expr") == "var88=sqrt(sq(var165)+sq(var166))"
assert src.grid() == "point"
assert src.spatial_dims == 2
def test_create_from_expr6():
src = ifs_source.read("88", "(var144==0)*sqrt(sq(var131)+sq(var132))")
eq_(src.get_grib_code(), grib_code(88, 128))
eq_(src.get_root_codes(),
[grib_code(144, 128),
grib_code(131, 128),
grib_code(132, 128)])
eq_(getattr(src, "expr"),
"var88=(var144==0)*sqrt(sq(var131)+sq(var132))")
eq_(src.spatial_dims, 3)
def test_create_from_expr4():
src = ifs_source.read("var129124",
"var126094 + var126099 + var126106 + var126110")
assert src.get_grib_code() == grib_code(124, 129)
assert set(src.get_root_codes()) == {
grib_code(94, 126),
grib_code(99, 126),
grib_code(106, 126),
grib_code(110, 126)
}
assert getattr(src, "expr") == "var124=var94+var99+var106+var110"
assert src.grid() == "point"
assert src.spatial_dims == 3
def test_create_from_expr4():
src = ifs_source.read("var129124",
"var126094 + var126099 + var126106 + var126110")
eq_(src.get_grib_code(), grib_code(124, 129))
eq_(src.get_root_codes(), [
grib_code(94, 126),
grib_code(99, 126),
grib_code(106, 126),
grib_code(110, 126)
])
eq_(getattr(src, "expr"), "var124=var94+var99+var106+var110")
eq_(src.grid(), "point")
eq_(src.spatial_dims, 3)
def test_create_from_expr_masked():
src = ifs_source.read("88",
"(var144==0)*sqrt(sq(var131)+sq(var132))",
mask_expr="var172>0.5")
assert src.get_grib_code() == grib_code(88, 128)
assert set(src.get_root_codes()) == {
grib_code(144, 128),
grib_code(131, 128),
grib_code(132, 128),
grib_code(172, 128)
}
assert getattr(
src, "expr") == "var88=(var144==0)*sqrt(sq(var131)+sq(var132))"
assert src.spatial_dims == 3
def test_invalid_expression2():
with LogCapture() as logc:
expr = "var89=sqrt(sq(var88)+sq(var166))"
src = ifs_source.read(expr)
ok_("cmor_source ERROR" in str(logc))
def test_create_from_string():
src = ifs_source.read("133.128")
eq_(src.get_grib_code(), grib_code(133, 128))
def test_create_from_var_string_masked():
src = ifs_source.read("var133", mask_expr="var172<=0.5")
assert set(src.get_root_codes()) == {
grib_code(133, 128), grib_code(172, 128)
}
def test_create_from_var_string():
src = ifs_source.read("var133")
assert src.get_grib_code() == grib_code(133, 128)
def test_invalid_expression2():
with LogCapture() as logc:
ifs_source.read("89.128", "sqrt(sq(var88)+sq(var166))")
assert "cmor_source ERROR" in str(logc)
def test_invalid_expression1():
with LogCapture() as logc:
src = ifs_source.read("141.128", "sqrt(sq(var165)+sq(var166))")
ok_("cmor_source ERROR" in str(logc))
