Skip to content

Commit 5dadfe4

Browse files
committed
Rename variable for consistency
1 parent 0a7c014 commit 5dadfe4

File tree

1 file changed

+20
-20
lines changed

1 file changed

+20
-20
lines changed

tests/test_tools.py

Lines changed: 20 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -8,58 +8,58 @@
88
class TestTools:
99

1010
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': [
1414
pd.Series(data=[4.0, 5.0, 6.0]),
1515
pd.Series(data=[8.0, 10.0, 14.0]),
1616
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)
2222
parameters = {'comp_value': 100,
2323
'column_name': 'v_wind'}
2424

2525
# comparative value is an index of data frame
2626
exp_output = (100, 4.0)
27-
assert smallest_difference(data_frame, **parameters) == exp_output
27+
assert smallest_difference(weather, **parameters) == exp_output
2828
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,
3030
**parameters)[1], exp_series)
3131
exp_arr = np.array(exp_series)
32-
assert_array_equal(smallest_difference(data_frame_arr,
32+
assert_array_equal(smallest_difference(weather_arr,
3333
**parameters)[1], exp_arr)
3434
# comparative value between indices of data frame
3535
exp_output = (150, 5.0)
3636
parameters['comp_value'] = 175
37-
assert smallest_difference(data_frame, **parameters) == exp_output
37+
assert smallest_difference(weather, **parameters) == exp_output
3838
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,
4040
**parameters)[1], exp_series)
4141
exp_arr = np.array(exp_series)
42-
assert_array_equal(smallest_difference(data_frame_arr,
42+
assert_array_equal(smallest_difference(weather_arr,
4343
**parameters)[1], exp_arr)
4444
# comparative value > indices of data frame
4545
exp_output = (200, 6.0)
4646
parameters['comp_value'] = 250
47-
assert smallest_difference(data_frame, **parameters) == exp_output
47+
assert smallest_difference(weather, **parameters) == exp_output
4848
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,
5050
**parameters)[1], exp_series)
5151
exp_arr = np.array(exp_series)
52-
assert_array_equal(smallest_difference(data_frame_arr,
52+
assert_array_equal(smallest_difference(weather_arr,
5353
**parameters)[1], exp_arr)
5454
# comparative value < indices of data frame
5555
exp_output = (100, 4.0)
5656
parameters['comp_value'] = 90
57-
assert smallest_difference(data_frame, **parameters) == exp_output
57+
assert smallest_difference(weather, **parameters) == exp_output
5858
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,
6060
**parameters)[1], exp_series)
6161
exp_arr = np.array(exp_series)
62-
assert_array_equal(smallest_difference(data_frame_arr,
62+
assert_array_equal(smallest_difference(weather_arr,
6363
**parameters)[1], exp_arr)
6464

6565
def test_linear_extra_interpolation(self):

0 commit comments

Comments
 (0)