|
8 | 8 | class TestTools: |
9 | 9 |
|
10 | 10 | def test_smallest_difference(self): |
11 | | - data_frame = pd.DataFrame(data={'v_wind': [4.0, 5.0, 6.0]}, |
12 | | - index=[100, 150, 200]) |
13 | | - data_frame_series = pd.DataFrame(data={'v_wind': [ |
| 11 | + weather = pd.DataFrame(data={'v_wind': [4.0, 5.0, 6.0]}, |
| 12 | + index=[100, 150, 200]) |
| 13 | + weather_series = pd.DataFrame(data={'v_wind': [ |
14 | 14 | pd.Series(data=[4.0, 5.0, 6.0]), |
15 | 15 | pd.Series(data=[8.0, 10.0, 14.0]), |
16 | 16 | pd.Series(data=[16.0, 20.0, 28.0])]}, index=[100, 150, 200]) |
17 | | - data_frame_arr = pd.DataFrame(data={'v_wind': [ |
18 | | - np.array(data_frame_series['v_wind'][data_frame_series.index[0]]), |
19 | | - np.array(data_frame_series['v_wind'][data_frame_series.index[1]]), |
20 | | - np.array(data_frame_series['v_wind'][data_frame_series.index[2]])]}, |
21 | | - index=data_frame_series.index) |
| 17 | + weather_arr = pd.DataFrame(data={'v_wind': [ |
| 18 | + np.array(weather_series['v_wind'][weather_series.index[0]]), |
| 19 | + np.array(weather_series['v_wind'][weather_series.index[1]]), |
| 20 | + np.array(weather_series['v_wind'][weather_series.index[2]])]}, |
| 21 | + index=weather_series.index) |
22 | 22 | parameters = {'comp_value': 100, |
23 | 23 | 'column_name': 'v_wind'} |
24 | 24 |
|
25 | 25 | # comparative value is an index of data frame |
26 | 26 | exp_output = (100, 4.0) |
27 | | - assert smallest_difference(data_frame, **parameters) == exp_output |
| 27 | + assert smallest_difference(weather, **parameters) == exp_output |
28 | 28 | exp_series = pd.Series(data=[4.0, 5.0, 6.0]) |
29 | | - assert_series_equal(smallest_difference(data_frame_series, |
| 29 | + assert_series_equal(smallest_difference(weather_series, |
30 | 30 | **parameters)[1], exp_series) |
31 | 31 | exp_arr = np.array(exp_series) |
32 | | - assert_array_equal(smallest_difference(data_frame_arr, |
| 32 | + assert_array_equal(smallest_difference(weather_arr, |
33 | 33 | **parameters)[1], exp_arr) |
34 | 34 | # comparative value between indices of data frame |
35 | 35 | exp_output = (150, 5.0) |
36 | 36 | parameters['comp_value'] = 175 |
37 | | - assert smallest_difference(data_frame, **parameters) == exp_output |
| 37 | + assert smallest_difference(weather, **parameters) == exp_output |
38 | 38 | exp_series = pd.Series(data=[8.0, 10.0, 14.0]) |
39 | | - assert_series_equal(smallest_difference(data_frame_series, |
| 39 | + assert_series_equal(smallest_difference(weather_series, |
40 | 40 | **parameters)[1], exp_series) |
41 | 41 | exp_arr = np.array(exp_series) |
42 | | - assert_array_equal(smallest_difference(data_frame_arr, |
| 42 | + assert_array_equal(smallest_difference(weather_arr, |
43 | 43 | **parameters)[1], exp_arr) |
44 | 44 | # comparative value > indices of data frame |
45 | 45 | exp_output = (200, 6.0) |
46 | 46 | parameters['comp_value'] = 250 |
47 | | - assert smallest_difference(data_frame, **parameters) == exp_output |
| 47 | + assert smallest_difference(weather, **parameters) == exp_output |
48 | 48 | exp_series = pd.Series(data=[16.0, 20.0, 28.0]) |
49 | | - assert_series_equal(smallest_difference(data_frame_series, |
| 49 | + assert_series_equal(smallest_difference(weather_series, |
50 | 50 | **parameters)[1], exp_series) |
51 | 51 | exp_arr = np.array(exp_series) |
52 | | - assert_array_equal(smallest_difference(data_frame_arr, |
| 52 | + assert_array_equal(smallest_difference(weather_arr, |
53 | 53 | **parameters)[1], exp_arr) |
54 | 54 | # comparative value < indices of data frame |
55 | 55 | exp_output = (100, 4.0) |
56 | 56 | parameters['comp_value'] = 90 |
57 | | - assert smallest_difference(data_frame, **parameters) == exp_output |
| 57 | + assert smallest_difference(weather, **parameters) == exp_output |
58 | 58 | exp_series = pd.Series(data=[4.0, 5.0, 6.0]) |
59 | | - assert_series_equal(smallest_difference(data_frame_series, |
| 59 | + assert_series_equal(smallest_difference(weather_series, |
60 | 60 | **parameters)[1], exp_series) |
61 | 61 | exp_arr = np.array(exp_series) |
62 | | - assert_array_equal(smallest_difference(data_frame_arr, |
| 62 | + assert_array_equal(smallest_difference(weather_arr, |
63 | 63 | **parameters)[1], exp_arr) |
64 | 64 |
|
65 | 65 | def test_linear_extra_interpolation(self): |
|
0 commit comments