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