81.818false87300clubb_jax.pytests.auto_llm_generated_pytests.test_adg1_adg2_responder_params0.8440test_responder_params_matches_referencefalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_adg1_adg2_responder_params0.0030test_alpha_x_clip_to_unit_intervalfalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_adg1_adg2_responder_params1.0960test_responder_params_grad_finitefalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_adg2_driver1.9010test_f2py_oraclefalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_adg2_driver3.460test_differentiablefalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_advance_helper_extras0.1050test_smooth_min_closed_formfalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_advance_helper_extras0.1230test_smooth_min_f2pyfalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_advance_helper_extras0.1640test_smooth_max_f2pyfalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_advance_helper_extras0.3680test_compute_Cx_fnc_Richardson_f2pyfalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_advance_helper_extras0.4040test_calc_xpwp_f2pyfalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_advance_helper_extras0.130test_calc_xpwp_identityfalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_advance_helper_extras1.0290test_differentiablefalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_advance_helper_extras0.2250test_stability_and_Ri_f2pyfalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_advance_helper_extras0.1220test_vertical_avg_integral_f2pyfalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_advance_helper_extras1.2840test_brunt_vaisala_f2pyfalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_advance_helper_extras0.2040test_calculate_thlp2_rad_f2pyfalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_advance_helper_extras0.1210test_Lscale_width_vert_avg_f2pyfalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_advance_helper_extras0.1630test_wp23_term_splat_lhs_f2pyfalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_advance_xp3_simplified0.6860test_term_rhs_formulasfalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_advance_xp3_simplified0.5530test_steady_state_vs_compositionfalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_binormal_moments0.3450test_f2py_oraclefalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_binormal_moments0.6120test_monte_carlofalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_binormal_moments0.6090test_differentiablefalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_bivar_LL_covar_partial0.1110test_general_partial_closed_form_and_mean_identityfalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_bivar_LL_covar_partial0.0550test_const_x2_partial_closed_form_and_limitfalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_bivar_LL_covar_partial0.5850test_grad_finitefalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_bivar_LL_covar_partial_rr_Nr0.3270test_rr_all_vary_generalfalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_bivar_LL_covar_partial_rr_Nr0.0020test_rr_x1_const_reuses_mean_const_x1falsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_bivar_LL_covar_partial_rr_Nr0.0010test_rr_x2_const_partialfalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_bivar_LL_covar_partial_rr_Nr0.0010test_rr_both_const_allfalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_bivar_LL_covar_partial_rr_Nr0.0010test_Nr_all_vary_swappedfalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_bivar_LL_mean_const0.10test_const_all_closed_form_and_limitfalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_bivar_LL_mean_const0.0540test_const_x1_closed_form_and_limitfalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_bivar_LL_mean_const0.4770test_grad_finitefalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_bivar_NL_mean_eq2.10test_both_vary_selects_generalfalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_bivar_NL_mean_eq0.0060test_x1_const_selects_const_x1falsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_bivar_NL_mean_eq0.0070test_x2_const_selects_const_x2falsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_bivar_NL_mean_eq0.0060test_both_const_selects_const_allfalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_bivar_NL_mean_eq0.0050test_parab_cyl_overflow_forces_const_x1falsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_bugsrad0.9590test_planck_vs_fortran_replicafalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_bugsrad0.1930test_newexp_vs_fortran_replicafalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_bugsrad0.7110test_rayle_vs_fortran_replicafalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_bugsrad1.6070test_gascon_vs_fortran_replicafalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_bugsrad1.9090test_cloudg_vs_fortran_replicafalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_bugsrad0.4430test_comscp_vs_fortran_replicafalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_bugsrad1.8760test_two_rt_lw_vs_fortran_replicafalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_bugsrad1.4320test_two_rt_sw_vs_fortran_replicafalsePASSED1clubb_jax.pytests.auto_llm_generated_pytests.test_bugsrad0.004FileNotFoundError: [Errno 2] No such file or directory: '/home/jenkins/workspace/clubb_branch_jax@2/clubb_release/src/Radiation/BUGSrad/gases_ckd_data.h'def test_gases_ckd_data_parser(): """The gases_ckd_data.h parser builds the correlated-k tables with correct shapes, fully filled (no missing T-coefficient slices), and the right spot-checked values.""" from clubb_jax.src.Radiation.BUGSrad.gases_ckd_data import tables, KG, NUMPS, NUMPIR, NUMTS, NUMTIR > t = tables() ^^^^^^^^ clubb_jax/pytests/auto_llm_generated_pytests/test_bugsrad.py:367: _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ clubb_jax/src/Radiation/BUGSrad/gases_ckd_data.py:102: in tables _TABLES = load_tables() ^^^^^^^^^^^^^ clubb_jax/src/Radiation/BUGSrad/gases_ckd_data.py:64: in load_tables for stmt in _logical_lines(os.path.normpath(_DATA_H)): ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ path = '/home/jenkins/workspace/clubb_branch_jax@2/clubb_release/src/Radiation/BUGSrad/gases_ckd_data.h' def _logical_lines(path): """Strip comments + #directives, join `&`-continued lines into logical statements.""" out, buf = [], '' > for raw in open(path): ^^^^^^^^^^ E FileNotFoundError: [Errno 2] No such file or directory: '/home/jenkins/workspace/clubb_branch_jax@2/clubb_release/src/Radiation/BUGSrad/gases_ckd_data.h' clubb_jax/src/Radiation/BUGSrad/gases_ckd_data.py:39: FileNotFoundError21test_gases_ckd_data_parserfalseFAILED1clubb_jax.pytests.auto_llm_generated_pytests.test_bugsrad0.0FileNotFoundError: [Errno 2] No such file or directory: '/home/jenkins/workspace/clubb_branch_jax@2/clubb_release/src/Radiation/BUGSrad/gases_ckd_data.h'def test_gases_dispatch_vs_fortran_replica(): """The `gases` 18-band dispatch (table indexing + overlap formulas + hk weights) vs a NumPy replica.""" from clubb_jax.src.Radiation.BUGSrad.gases_ckd import gases, pscale from clubb_jax.src.Radiation.BUGSrad.gases_ckd_data import tables, KG, STANPS, STANPIR from clubb_jax.src.Radiation.BUGSrad.bugsrad_physconst import MOLAR_VOLUME as MV, GRAVITY as G, \ MW_H2O as MWh, MW_O3 as MWo, MW_DRY_AIR as MWd > T = tables() ^^^^^^^^ clubb_jax/pytests/auto_llm_generated_pytests/test_bugsrad.py:396: _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ clubb_jax/src/Radiation/BUGSrad/gases_ckd_data.py:102: in tables _TABLES = load_tables() ^^^^^^^^^^^^^ clubb_jax/src/Radiation/BUGSrad/gases_ckd_data.py:64: in load_tables for stmt in _logical_lines(os.path.normpath(_DATA_H)): ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ path = '/home/jenkins/workspace/clubb_branch_jax@2/clubb_release/src/Radiation/BUGSrad/gases_ckd_data.h' def _logical_lines(path): """Strip comments + #directives, join `&`-continued lines into logical statements.""" out, buf = [], '' > for raw in open(path): ^^^^^^^^^^ E FileNotFoundError: [Errno 2] No such file or directory: '/home/jenkins/workspace/clubb_branch_jax@2/clubb_release/src/Radiation/BUGSrad/gases_ckd_data.h' clubb_jax/src/Radiation/BUGSrad/gases_ckd_data.py:39: FileNotFoundError21test_gases_dispatch_vs_fortran_replicafalseFAILED0clubb_jax.pytests.auto_llm_generated_pytests.test_bugsrad1.6370test_gases_ckd_helpers_vs_fortran_replicafalsePASSED1clubb_jax.pytests.auto_llm_generated_pytests.test_bugsrad4.458FileNotFoundError: [Errno 2] No such file or directory: '/home/jenkins/workspace/clubb_branch_jax@2/clubb_release/src/Radiation/BUGSrad/gases_ckd_data.h'def test_bugs_lwr(): """LW band driver: the no-cloud invariant (all-sky == clear-sky, bit-exact), finiteness, physical flux ranges, and clouds adding downward LW.""" from clubb_jax.src.Radiation.BUGSrad.bugs_lwr import bugs_lwr rng = np.random.default_rng(11) ncol, nlm = 2, 25 pp = np.cumsum(15.0 + 10.0 * rng.random((ncol, nlm + 1)), axis=1) + 1.0 dp = pp[:, 1:] - pp[:, :-1]; ppl = 0.5 * (pp[:, :-1] + pp[:, 1:]) tt = 220.0 + 70.0 * rng.random((ncol, nlm)); ts = 290.0 + 10.0 * rng.random(ncol) rmix = 1e-3 * rng.random((ncol, nlm)); o3mix = 1e-6 * rng.random((ncol, nlm)) cw0 = np.zeros((ncol, nlm)); ci0 = np.zeros((ncol, nlm)) fd, fu, fdcl, fucl = (np.asarray(x) for x in > bugs_lwr(ts, ppl, dp, tt, rmix, cw0, ci0, o3mix, pp, 330.0, 1.6, 0.28)) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ clubb_jax/pytests/auto_llm_generated_pytests/test_bugsrad.py:533: _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ clubb_jax/src/Radiation/BUGSrad/bugs_lwr.py:77: in bugs_lwr tg, hk = gases(ib, ig, dp, tt, rmix, o3mix, umco2, umch4, umn2o, pkd, ip1, ip2, pp) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ clubb_jax/src/Radiation/BUGSrad/gases_ckd.py:108: in gases T = tables() ^^^^^^^^ clubb_jax/src/Radiation/BUGSrad/gases_ckd_data.py:102: in tables _TABLES = load_tables() ^^^^^^^^^^^^^ clubb_jax/src/Radiation/BUGSrad/gases_ckd_data.py:64: in load_tables for stmt in _logical_lines(os.path.normpath(_DATA_H)): ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ path = '/home/jenkins/workspace/clubb_branch_jax@2/clubb_release/src/Radiation/BUGSrad/gases_ckd_data.h' def _logical_lines(path): """Strip comments + #directives, join `&`-continued lines into logical statements.""" out, buf = [], '' > for raw in open(path): ^^^^^^^^^^ E FileNotFoundError: [Errno 2] No such file or directory: '/home/jenkins/workspace/clubb_branch_jax@2/clubb_release/src/Radiation/BUGSrad/gases_ckd_data.h' clubb_jax/src/Radiation/BUGSrad/gases_ckd_data.py:39: FileNotFoundError21test_bugs_lwrfalseFAILED1clubb_jax.pytests.auto_llm_generated_pytests.test_bugsrad0.757FileNotFoundError: [Errno 2] No such file or directory: '/home/jenkins/workspace/clubb_branch_jax@2/clubb_release/src/Radiation/BUGSrad/gases_ckd_data.h'def test_bugs_swr(): """SW band driver: no-cloud invariant (all-sky==clear-sky bit-exact), the TOA incident solar (fdsw[TOA] = amu0·slr·Σband_solar = amu0·slr·1339.946) validating the accumulation, and finiteness.""" from clubb_jax.src.Radiation.BUGSrad.bugs_swr import bugs_swr rng = np.random.default_rng(12) ncol, nlm = 2, 25 pp = np.cumsum(15.0 + 10.0 * rng.random((ncol, nlm + 1)), axis=1) + 1.0 dp = pp[:, 1:] - pp[:, :-1]; ppl = 0.5 * (pp[:, :-1] + pp[:, 1:]) tt = 230.0 + 60.0 * rng.random((ncol, nlm)); ts = 290.0 * np.ones(ncol) rmix = 1e-3 * rng.random((ncol, nlm)); o3mix = 1e-6 * rng.random((ncol, nlm)) amu0 = np.array([0.5, 0.8]); slr = np.ones(ncol); alb = 0.2 * np.ones(ncol) cw0 = np.zeros((ncol, nlm)); ci0 = np.zeros((ncol, nlm)) out = [np.asarray(x) for x in > bugs_swr(ts, amu0, slr, alb, alb, alb, alb, ppl, dp, tt, rmix, cw0, ci0, o3mix, pp, 330.0, 1.6, 0.28)] ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ clubb_jax/pytests/auto_llm_generated_pytests/test_bugsrad.py:558: _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ clubb_jax/src/Radiation/BUGSrad/bugs_swr.py:65: in bugs_swr tg, hk = gases(ib, ig, dp, ttem, rmix, o3mix, umco2, umch4, umn2o, pkd, ip1, ip2, pp) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ clubb_jax/src/Radiation/BUGSrad/gases_ckd.py:108: in gases T = tables() ^^^^^^^^ clubb_jax/src/Radiation/BUGSrad/gases_ckd_data.py:102: in tables _TABLES = load_tables() ^^^^^^^^^^^^^ clubb_jax/src/Radiation/BUGSrad/gases_ckd_data.py:64: in load_tables for stmt in _logical_lines(os.path.normpath(_DATA_H)): ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ path = '/home/jenkins/workspace/clubb_branch_jax@2/clubb_release/src/Radiation/BUGSrad/gases_ckd_data.h' def _logical_lines(path): """Strip comments + #directives, join `&`-continued lines into logical statements.""" out, buf = [], '' > for raw in open(path): ^^^^^^^^^^ E FileNotFoundError: [Errno 2] No such file or directory: '/home/jenkins/workspace/clubb_branch_jax@2/clubb_release/src/Radiation/BUGSrad/gases_ckd_data.h' clubb_jax/src/Radiation/BUGSrad/gases_ckd_data.py:39: FileNotFoundError21test_bugs_swrfalseFAILED1clubb_jax.pytests.auto_llm_generated_pytests.test_bugsrad4.462FileNotFoundError: [Errno 2] No such file or directory: '/home/jenkins/workspace/clubb_branch_jax@2/clubb_release/src/Radiation/BUGSrad/gases_ckd_data.h'def test_bugs_rad(): """Main orchestration: LW heating conserves (telescoping flux divergence), night columns get zero SW heating, the no-cloud invariant (all-sky heating == clear-sky), and physical heating rates.""" from clubb_jax.src.Radiation.BUGSrad.bugs_rad import bugs_rad from clubb_jax.src.Radiation.BUGSrad.bugsrad_physconst import GRAVITY, CP_DRY_AIR rng = np.random.default_rng(13) ncol, nlm = 2, 30 pl2 = np.cumsum(15.0 + 10.0 * rng.random((ncol, nlm + 1)), axis=1) + 1.0 dpl = pl2[:, 1:] - pl2[:, :-1]; ppl = 0.5 * (pl2[:, :-1] + pl2[:, 1:]) tl = 220.0 + 70.0 * rng.random((ncol, nlm)); ts = 290.0 * np.ones(ncol) ql = 1e-3 * rng.random((ncol, nlm)); o3 = 1e-6 * rng.random((ncol, nlm)) # specific humidity, ozone amu0 = np.array([0.5, 0.005]); slr = np.ones(ncol); alb = 0.2 * np.ones(ncol) # col 0 day, col 1 night qcwl = np.where(rng.random((ncol, nlm)) > 0.6, 1e-4, 0.0) # cloud-water mixing ratio [kg/kg] qcil = np.zeros((ncol, nlm)); acld = np.where(qcwl > 0, 1.0, 0.0) # cloud ice, cloud fraction > r = bugs_rad(pl2, ppl, dpl, tl, ql, qcwl, qcil, o3, ts, amu0, slr, alb, alb, alb, alb, acld, s0=1367.0) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ clubb_jax/pytests/auto_llm_generated_pytests/test_bugsrad.py:582: _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ clubb_jax/src/Radiation/BUGSrad/bugs_rad.py:62: in bugs_rad fdlw, fulw, fdlwcl, fulwcl = bugs_lwr(ts, ppl, dpl, tl, rmix, cwrho, cirho, o3mix, pl2, clubb_jax/src/Radiation/BUGSrad/bugs_lwr.py:77: in bugs_lwr tg, hk = gases(ib, ig, dp, tt, rmix, o3mix, umco2, umch4, umn2o, pkd, ip1, ip2, pp) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ clubb_jax/src/Radiation/BUGSrad/gases_ckd.py:108: in gases T = tables() ^^^^^^^^ clubb_jax/src/Radiation/BUGSrad/gases_ckd_data.py:102: in tables _TABLES = load_tables() ^^^^^^^^^^^^^ clubb_jax/src/Radiation/BUGSrad/gases_ckd_data.py:64: in load_tables for stmt in _logical_lines(os.path.normpath(_DATA_H)): ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ path = '/home/jenkins/workspace/clubb_branch_jax@2/clubb_release/src/Radiation/BUGSrad/gases_ckd_data.h' def _logical_lines(path): """Strip comments + #directives, join `&`-continued lines into logical statements.""" out, buf = [], '' > for raw in open(path): ^^^^^^^^^^ E FileNotFoundError: [Errno 2] No such file or directory: '/home/jenkins/workspace/clubb_branch_jax@2/clubb_release/src/Radiation/BUGSrad/gases_ckd_data.h' clubb_jax/src/Radiation/BUGSrad/gases_ckd_data.py:39: FileNotFoundError21test_bugs_radfalseFAILED1clubb_jax.pytests.auto_llm_generated_pytests.test_bugsrad1.935FileNotFoundError: [Errno 2] No such file or directory: '/home/jenkins/workspace/clubb_branch_jax@2/clubb_release/src/Radiation/BUGSrad/gases_ckd_data.h' -------------------- For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.def test_bugsrad_driver(): """CLUBB↔BUGSrad interface: std-atmosphere loads (50 levels), determine_extended_atmos_bounds gives sane bounds for a gabls3-like grid (model top 5 km, radiation_top 30 km), and compute_bugsrad_radiation produces a strictly-increasing radiation-grid pressure, finite CLUBB-grid heating, and zero SW at night.""" import numpy as np from clubb_jax.src.Input_fields.sounding import load_extended_std_atm from clubb_jax.src.Radiation.bugsrad_driver import ( determine_extended_atmos_bounds, build_rad_grid_setup, compute_bugsrad_radiation) ext = load_extended_std_atm() assert ext['alt'].shape == (50,) and ext['alt'][0] == 1000.0 and ext['alt'][-1] == 50000.0 nzm, dz, ncol = 51, 100.0, 2 zm = np.arange(nzm) * dz # 0..5000 m zm_spacing = np.full(nzm - 1, dz) p_zm = 1013e2 * np.exp(-zm / 8000.0) # Pa, hydrostatic (top ≈ 542 mb) rad_top = 30000.0 bottom, top, rsize, linbuf = determine_extended_atmos_bounds(zm, zm_spacing, p_zm, rad_top, ext) assert rsize >= 1 and linbuf >= 0 and top >= bottom, "degenerate extension bounds" assert ext['alt'][bottom - 1] >= 5000.0 and ext['alt'][top - 1] <= 30000.0, "bounds outside [5,30] km" setup = build_rad_grid_setup(zm, zm_spacing, p_zm, rad_top, ext) assert setup['buffer'] == linbuf + rsize # synthetic state on the nzt grid nzt = nzm - 1; rng = np.random.default_rng(7) zt = 0.5 * (zm[1:] + zm[:-1]) T = np.tile(288.0 - 6.5e-3 * zt, (ncol, 1)) p_zt = np.tile(1013e2 * np.exp(-zt / 8000.0), (ncol, 1)) exner = (p_zt / 1.0e5) ** (287.0 / 1004.0) rho = np.tile(p_zm / (287.0 * (288.0 - 6.5e-3 * zm)), (ncol, 1)) rtm = 5e-3 * np.ones((ncol, nzt)); rcm = np.zeros((ncol, nzt)); rcm[:, 5:8] = 1e-4 cf = np.where(rcm > 0, 1.0, 0.0); rim = np.zeros_like(rcm); rsm = np.zeros_like(rcm); isf = np.zeros_like(rcm) amu0 = np.array([0.5, 0.0]); slr = np.ones(ncol); alb = 0.23 * np.ones(ncol) p_zm2 = np.tile(p_zm, (ncol, 1)) > res = compute_bugsrad_radiation(setup, T, rcm, rtm, rsm, rim, cf, isf, p_zt, p_zm2, rho, exner, amu0, slr, alb, alb, alb, alb, sol_const=1321.0) clubb_jax/pytests/auto_llm_generated_pytests/test_bugsrad.py:628: _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ clubb_jax/src/Radiation/bugsrad_driver.py:143: in compute_bugsrad_radiation out = fn(playerinmb, p_rad, diff_pres, T_rad, q_rad, rcmic_rad, rcil_rad, o3_rad, clubb_jax/src/Radiation/BUGSrad/bugs_rad.py:62: in bugs_rad fdlw, fulw, fdlwcl, fulwcl = bugs_lwr(ts, ppl, dpl, tl, rmix, cwrho, cirho, o3mix, pl2, clubb_jax/src/Radiation/BUGSrad/bugs_lwr.py:77: in bugs_lwr tg, hk = gases(ib, ig, dp, tt, rmix, o3mix, umco2, umch4, umn2o, pkd, ip1, ip2, pp) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ clubb_jax/src/Radiation/BUGSrad/gases_ckd.py:108: in gases T = tables() ^^^^^^^^ clubb_jax/src/Radiation/BUGSrad/gases_ckd_data.py:102: in tables _TABLES = load_tables() ^^^^^^^^^^^^^ clubb_jax/src/Radiation/BUGSrad/gases_ckd_data.py:64: in load_tables for stmt in _logical_lines(os.path.normpath(_DATA_H)): ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ path = '/home/jenkins/workspace/clubb_branch_jax@2/clubb_release/src/Radiation/BUGSrad/gases_ckd_data.h' def _logical_lines(path): """Strip comments + #directives, join `&`-continued lines into logical statements.""" out, buf = [], '' > for raw in open(path): ^^^^^^^^^^ E FileNotFoundError: [Errno 2] No such file or directory: '/home/jenkins/workspace/clubb_branch_jax@2/clubb_release/src/Radiation/BUGSrad/gases_ckd_data.h' E -------------------- E For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these. clubb_jax/src/Radiation/BUGSrad/gases_ckd_data.py:39: FileNotFoundError21test_bugsrad_driverfalseFAILED1clubb_jax.pytests.auto_llm_generated_pytests.test_bugsrad0.084TypeError: Error interpreting argument to <function advance_clubb_radiation at 0x727c3a173a60> as an abstract array. The problematic value is of type <class 'str'> and was passed to the function at path args[0]['cfg']['rad_scheme']. This typically means that a jit-wrapped function was called with a non-array argument, and this argument was not marked as static using the static_argnums or static_argnames parameters of jax.jit. -------------------- For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.def test_bugsrad_radiation_dispatch(): """End-to-end wiring: advance_clubb_radiation(rad_scheme='bugsrad') builds the grid setup, maps a gabls3-like CLUBB state through bugs_rad, and writes a finite K/s heating rate into state['radht'] (shape (1,nzt)).""" import numpy as np from clubb_jax.src.CLUBB_core.grid_class import setup_grid from clubb_jax.src.Radiation.radiation_module import advance_clubb_radiation gr = setup_grid(ngrdcol=1, deltaz=100.0, zm_init=0.0, zm_top=5000.0) nzt, nzm = gr.nzt, gr.nzm zt = np.asarray(gr.zt)[0] p_zt = (1013e2 * np.exp(-zt / 8000.0))[None, :] exner = (p_zt / 1.0e5) ** (287.0 / 1004.67) thlm = np.full((1, nzt), 295.0); rtm = np.full((1, nzt), 5e-3) rcm = np.zeros((1, nzt)); rcm[:, 5:9] = 2e-4 cloud_frac = np.where(rcm > 0, 1.0, 0.0) rho_zm = (np.asarray(p_zt) / (287.0 * 290.0)) rho_zm = np.concatenate([rho_zm, rho_zm[:, -1:]], axis=1) # pad to nzm cfg = dict(rad_scheme="bugsrad", radiation_top=30000.0, sol_const=1321.0, slr=1.0, alvdr=0.23, alvdf=0.23, alndr=0.23, alndf=0.23, l_fix_cos_solar_zen=True, cos_solar_zen_values=[0.5]) state = dict(cfg=cfg, gr=gr, ngrdcol=1, nzt=nzt, nzm=nzm, stats_writer=None, rad_scheme="bugsrad", p_in_Pa=p_zt, exner=exner, thlm=thlm, rtm=rtm, rcm=rcm, cloud_frac=cloud_frac, ice_supersat_frac=np.zeros((1, nzt)), rho_zm=rho_zm, radht=np.zeros((1, nzt))) > advance_clubb_radiation(state, time_current=43200.0) # local noon-ish ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ E TypeError: Error interpreting argument to <function advance_clubb_radiation at 0x727c3a173a60> as an abstract array. The problematic value is of type <class 'str'> and was passed to the function at path args[0]['cfg']['rad_scheme']. E This typically means that a jit-wrapped function was called with a non-array argument, and this argument was not marked as static using the static_argnums or static_argnames parameters of jax.jit. E -------------------- E For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these. clubb_jax/pytests/auto_llm_generated_pytests/test_bugsrad.py:661: TypeError21test_bugsrad_radiation_dispatchfalseFAILED1clubb_jax.pytests.auto_llm_generated_pytests.test_bugsrad5.764FileNotFoundError: [Errno 2] No such file or directory: '/home/jenkins/workspace/clubb_branch_jax@2/clubb_release/src/Radiation/BUGSrad/gases_ckd_data.h' -------------------- For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.def test_bugs_rad_differentiable(): """The BUGSrad radiative transfer is DIFFERENTIABLE (a stated project goal): jax.grad of a radiative loss (TOA outgoing LW + total SW heating) w.r.t. temperature and cloud water is finite AND nonzero (the radiation responds to its physical inputs). cloudg is now fully float64 (REFACTOR A3).""" import jax, jax.numpy as jnp from clubb_jax.src.Radiation.BUGSrad.bugs_rad import bugs_rad rng = np.random.default_rng(1); ncol, nlm = 1, 40 pl2 = jnp.asarray(np.cumsum(15.0 + 10.0 * rng.random((ncol, nlm + 1)), axis=1) + 1.0) dpl = pl2[:, 1:] - pl2[:, :-1]; ppl = 0.5 * (pl2[:, :-1] + pl2[:, 1:]) tl0 = jnp.asarray(220.0 + 70.0 * rng.random((ncol, nlm))); ts = jnp.asarray(290.0 * np.ones(ncol)) ql = jnp.asarray(1e-3 * rng.random((ncol, nlm))); o3 = jnp.asarray(1e-6 * rng.random((ncol, nlm))) qcwl0 = jnp.asarray(np.where(rng.random((ncol, nlm)) > 0.5, 1e-4, 0.0)); qcil = jnp.zeros((ncol, nlm)) acld = jnp.asarray(np.where(np.asarray(qcwl0) > 0, 1.0, 0.0)) amu0 = jnp.asarray([0.5]); slr = jnp.ones(ncol); alb = jnp.asarray(0.2 * np.ones(ncol)) def loss(tl, qcwl): r = bugs_rad(pl2, ppl, dpl, tl, ql, qcwl, qcil, o3, ts, amu0, slr, alb, alb, alb, alb, acld, s0=1367.0) return jnp.sum(r['fulw'][:, 0]) + jnp.sum(r['asl']) > g_t, g_q = jax.grad(loss, argnums=(0, 1))(tl0, qcwl0) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ clubb_jax/pytests/auto_llm_generated_pytests/test_bugsrad.py:691: _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ clubb_jax/pytests/auto_llm_generated_pytests/test_bugsrad.py:688: in loss r = bugs_rad(pl2, ppl, dpl, tl, ql, qcwl, qcil, o3, ts, amu0, slr, alb, alb, alb, alb, acld, s0=1367.0) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ clubb_jax/src/Radiation/BUGSrad/bugs_rad.py:62: in bugs_rad fdlw, fulw, fdlwcl, fulwcl = bugs_lwr(ts, ppl, dpl, tl, rmix, cwrho, cirho, o3mix, pl2, clubb_jax/src/Radiation/BUGSrad/bugs_lwr.py:77: in bugs_lwr tg, hk = gases(ib, ig, dp, tt, rmix, o3mix, umco2, umch4, umn2o, pkd, ip1, ip2, pp) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ clubb_jax/src/Radiation/BUGSrad/gases_ckd.py:108: in gases T = tables() ^^^^^^^^ clubb_jax/src/Radiation/BUGSrad/gases_ckd_data.py:102: in tables _TABLES = load_tables() ^^^^^^^^^^^^^ clubb_jax/src/Radiation/BUGSrad/gases_ckd_data.py:64: in load_tables for stmt in _logical_lines(os.path.normpath(_DATA_H)): ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ path = '/home/jenkins/workspace/clubb_branch_jax@2/clubb_release/src/Radiation/BUGSrad/gases_ckd_data.h' def _logical_lines(path): """Strip comments + #directives, join `&`-continued lines into logical statements.""" out, buf = [], '' > for raw in open(path): ^^^^^^^^^^ E FileNotFoundError: [Errno 2] No such file or directory: '/home/jenkins/workspace/clubb_branch_jax@2/clubb_release/src/Radiation/BUGSrad/gases_ckd_data.h' E -------------------- E For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these. clubb_jax/src/Radiation/BUGSrad/gases_ckd_data.py:39: FileNotFoundError21test_bugs_rad_differentiablefalseFAILED0clubb_jax.pytests.auto_llm_generated_pytests.test_calc_comp_corrs_binormal1.2240test_f2py_oraclefalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_calc_comp_corrs_binormal1.3630test_round_tripfalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_calc_comp_corrs_binormal0.0030test_boundfalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_calc_comp_corrs_binormal0.6420test_smooth_corr_quotient_f2pyfalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_calc_comp_corrs_binormal1.3390test_differentiablefalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_calc_comp_mu_sigma_hm2.0450test_both_precip_preserves_mean_and_variancefalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_calc_comp_mu_sigma_hm0.0110test_rico_params_non_emergencyfalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_calc_comp_mu_sigma_hm0.0150test_root_sign_selectionfalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_calc_comp_mu_sigma_hm0.0070test_emergency_preserves_meanfalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_calc_comp_mu_sigma_hm0.0230test_single_component_branchesfalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_calc_comp_mu_sigma_hm2.2720test_no_precip_and_differentiablefalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_calc_comp_mu_sigma_hm1.8090test_compute_mean_stdev_columnsfalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_calc_comp_mu_sigma_hm0.2630test_norm_transform_matches_L2NfalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_calc_comp_mu_sigma_hm2.7850test_orchestration_differentiablefalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_calc_corr_norm_cholesky2.8470test_reconstruction_and_structurefalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_calc_corr_norm_cholesky0.0460test_adjustments_and_selectionfalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_calc_corr_norm_cholesky3.6750test_differentiablefalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_calc_corr_w_hm_n0.4180test_round_tripfalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_calc_corr_w_hm_n0.3340test_branches_vs_numpyfalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_calc_corr_w_hm_n0.0010test_clampfalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_calc_corr_w_hm_n0.8550test_differentiablefalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_calc_ice_cloud_frac_component3.3840test_below_freezing_closed_form_and_mcfalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_calc_ice_cloud_frac_component0.0050test_above_freezing_returns_cf_liqfalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_calc_ice_cloud_frac_component0.0050test_truncation_limitsfalsePASSED0clubb_jax.pytests.auto_llm_generated_pytests.test_calc_ice_cloud_frac_component0.7880test_grad_finitefalsePASSED81.818Declarative: Post Actions54pytest552026-10-04T18:36:15.669528-05:00