st-bender /
pyeppaurora
| @@ 96-116 (lines=21) @@ | ||
| 93 | return |
|
| 94 | ||
| 95 | ||
| 96 | @pytest.mark.parametrize( |
|
| 97 | "edissfunc, expected", |
|
| 98 | EDISS_FUNCS_EXPECTED, |
|
| 99 | ) |
|
| 100 | def test_endiss_3d_transposed(edissfunc, expected): |
|
| 101 | energies = np.logspace(-1, 2, 4) |
|
| 102 | fluxes = np.ones_like(energies) |
|
| 103 | # ca. 100, 150, 200 km |
|
| 104 | scale_heights = np.array([6e5, 27e5, 40e5]) |
|
| 105 | rhos = np.array([5e-10, 1.7e-12, 2.6e-13]) |
|
| 106 | ediss = edissfunc( |
|
| 107 | energies[None, :, None], fluxes[None, :, None], |
|
| 108 | scale_heights[:, None, None], rhos[:, None, None] |
|
| 109 | ) |
|
| 110 | assert ediss.shape == (3, 4, 1) |
|
| 111 | np.testing.assert_allclose(ediss[0, 2, 0], expected) |
|
| 112 | return |
|
| 113 | ||
| 114 | ||
| 115 | def test_ssusi_ioniz(): |
|
| 116 | energies = np.logspace(-1, 2, 4) |
|
| 117 | fluxes = np.ones_like(energies) |
|
| 118 | z = np.array([100, 120, 150]) |
|
| 119 | # energy dissipation "profiles" |
|
| @@ 77-97 (lines=21) @@ | ||
| 74 | return |
|
| 75 | ||
| 76 | ||
| 77 | @pytest.mark.parametrize( |
|
| 78 | "edissfunc, expected", |
|
| 79 | EDISS_FUNCS_EXPECTED, |
|
| 80 | ) |
|
| 81 | def test_endiss_3d(edissfunc, expected): |
|
| 82 | energies = np.logspace(-1, 2, 4) |
|
| 83 | fluxes = np.ones_like(energies) |
|
| 84 | # ca. 100, 150, 200 km |
|
| 85 | scale_heights = np.array([6e5, 27e5, 40e5]) |
|
| 86 | rhos = np.array([5e-10, 1.7e-12, 2.6e-13]) |
|
| 87 | ediss = edissfunc( |
|
| 88 | energies[None, None, :], fluxes[None, None, :], |
|
| 89 | scale_heights[:, None, None], rhos[:, None, None] |
|
| 90 | ) |
|
| 91 | assert ediss.shape == (3, 1, 4) |
|
| 92 | np.testing.assert_allclose(ediss[0, 0, 2], expected) |
|
| 93 | return |
|
| 94 | ||
| 95 | ||
| 96 | @pytest.mark.parametrize( |
|
| 97 | "edissfunc, expected", |
|
| 98 | EDISS_FUNCS_EXPECTED, |
|
| 99 | ) |
|
| 100 | def test_endiss_3d_transposed(edissfunc, expected): |
|
