| @@ 99-115 (lines=17) @@ | ||
| 96 | ) |
|
| 97 | ||
| 98 | ||
| 99 | @pytest.mark.parametrize( |
|
| 100 | "name, result", |
|
| 101 | [ |
|
| 102 | ("Ap", np.array([56, 39, 27, 18, 32, 15, 32, 22])), |
|
| 103 | ("Kp", np.array([5.3, 4.7, 4.0, 3.3, 4.3, 3.0, 4.3, 3.7])), |
|
| 104 | ] |
|
| 105 | ) |
|
| 106 | def test_3hourly_index(name, result, df_3h): |
|
| 107 | df = df_3h |
|
| 108 | np.testing.assert_allclose( |
|
| 109 | df.loc[ |
|
| 110 | pd.date_range( |
|
| 111 | "2000-01-01 01:30", "2000-01-01 23:30", freq="3h" |
|
| 112 | ) |
|
| 113 | ][name].values, |
|
| 114 | result, |
|
| 115 | rtol=1e-12, |
|
| 116 | ) |
|
| 117 | ||
| @@ 84-100 (lines=17) @@ | ||
| 81 | pd.testing.assert_series_equal(df1_ind, df2_ind) |
|
| 82 | ||
| 83 | ||
| 84 | @pytest.mark.parametrize( |
|
| 85 | "name, result", |
|
| 86 | [ |
|
| 87 | ("Ap", np.array([56, 39, 27, 18, 32, 15, 32, 22])), |
|
| 88 | ("Kp", np.array([5.3, 4.7, 4.0, 3.3, 4.3, 3.0, 4.3, 3.7])), |
|
| 89 | ] |
|
| 90 | ) |
|
| 91 | def test_3hourly_index(name, result, df_o): |
|
| 92 | df = df_o |
|
| 93 | np.testing.assert_allclose( |
|
| 94 | df.loc[ |
|
| 95 | pd.date_range( |
|
| 96 | "2000-01-01 00:00", "2000-01-01 23:00", freq="3h" |
|
| 97 | ) |
|
| 98 | ][name].values, |
|
| 99 | result, |
|
| 100 | rtol=1e-12, |
|
| 101 | ) |
|
| 102 | ||
| 103 | ||