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