Skip to content

Commit cb84935

Browse files
author
Giacomo Ferroni
committed
pylint checks
1 parent 26c86a5 commit cb84935

File tree

1 file changed

+13
-7
lines changed

1 file changed

+13
-7
lines changed

pandas/tests/test_window.py

Lines changed: 13 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -210,16 +210,22 @@ def f():
210210
def test_count_nonnumeric_types(self):
211211
# GH12541
212212
df_inf = DataFrame({'x': [1, 2, 3], 'y': [1., 2., np.Inf]})
213-
df_date = DataFrame({'x': [1, 2, 3], 'y': pd.date_range('20130101',periods=3)})
213+
df_date = DataFrame({'x': [1, 2, 3],
214+
'y': pd.date_range('20130101',periods=3)})
214215
df_inf_date = DataFrame({'x': [1, 2, 3], 'y': [1., 2., np.Inf],
215216
'z': pd.date_range('20170101',periods=3)})
216217

217-
expected_1 = DataFrame([[1,1],[2,2],[2,2]], columns=['x','y'], dtype=float)
218-
expected_2 = DataFrame([[1,1,1],[2,2,2],[2,2,2]], columns=['x','y','z'], dtype=float)
219-
220-
self.assert_frame_equal(df_inf.rolling(window=2).count(), expected_1)
221-
self.assert_frame_equal(df_date.rolling(window=2).count(), expected_1)
222-
self.assert_frame_equal(df_inf_date.rolling(window=2).count(), expected_2)
218+
expected_1 = DataFrame([[1,1],[2,2],[2,2]],
219+
columns=['x','y'], dtype=float)
220+
expected_2 = DataFrame([[1,1,1],[2,2,2],[2,2,2]],
221+
columns=['x','y','z'], dtype=float)
222+
223+
self.assert_frame_equal(df_inf.rolling(window=2).count(),
224+
expected_1)
225+
self.assert_frame_equal(df_date.rolling(window=2).count(),
226+
expected_1)
227+
self.assert_frame_equal(df_inf_date.rolling(window=2).count(),
228+
expected_2)
223229

224230
def test_window_with_args(self):
225231
tm._skip_if_no_scipy()

0 commit comments

Comments
 (0)