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))