Skip to content

Commit f117029

Browse files
committed
Fix minor issues with tests for Series.sort_index
1 parent 47b42ea commit f117029

File tree

1 file changed

+5
-6
lines changed

1 file changed

+5
-6
lines changed

pandas/tests/series/test_sorting.py

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -148,15 +148,14 @@ def test_sort_index_multiindex(self):
148148
def test_sort_index_nan(self):
149149

150150
# GH13729
151-
nan = np.nan
152-
ser = Series(['A', nan, 'C', 'D'], [1, 2, 0, nan])
151+
ser = Series(['A', np.nan, 'C', 'D'], [1, 2, 0, np.nan])
153152

154153
# na_position='last', kind='quicksort'
155154
sorted_series = ser.sort_index(kind='quicksort', na_position='last')
156-
expected_series = Series(['C', 'A', nan, 'D'], [0, 1, 2, nan])
155+
expected_series = Series(['C', 'A', np.nan, 'D'], [0, 1, 2, np.nan])
157156
assert_series_equal(sorted_series, expected_series)
158157

159158
# na_position='first'
160-
sorted_series = ser.sort_index(na_position='first')
161-
expected_series = Series(['D', 'C', 'A', nan], [nan, 0, 1, 2])
162-
assert_series_equal(sorted_series, expected_series)
159+
sorted = ser.sort_index(na_position='first')
160+
expected = Series(['D', 'C', 'A', np.nan], [np.nan, 0, 1, 2])
161+
assert_series_equal(sorted, expected)

0 commit comments

Comments
 (0)