33import numpy as np
44import pytest
55
6+ import pandas .util ._test_decorators as td
7+
68from pandas import DataFrame , Series
79import pandas .util .testing as tm
810
@@ -26,8 +28,10 @@ def method(self, request):
2628 """
2729 return request .param
2830
31+ @td .skip_if_no_scipy
2932 def test_rank (self , float_frame ):
30- rankdata = pytest .importorskip ("scipy.stats.rankdata" )
33+ import scipy .stats # noqa:F401
34+ from scipy .stats import rankdata
3135
3236 float_frame ["A" ][::2 ] = np .nan
3337 float_frame ["B" ][::3 ] = np .nan
@@ -117,8 +121,10 @@ def test_rank_mixed_frame(self, float_string_frame):
117121 expected = float_string_frame .rank (1 , numeric_only = True )
118122 tm .assert_frame_equal (result , expected )
119123
124+ @td .skip_if_no_scipy
120125 def test_rank_na_option (self , float_frame ):
121- rankdata = pytest .importorskip ("scipy.stats.rankdata" )
126+ import scipy .stats # noqa:F401
127+ from scipy .stats import rankdata
122128
123129 float_frame ["A" ][::2 ] = np .nan
124130 float_frame ["B" ][::3 ] = np .nan
@@ -199,9 +205,10 @@ def test_rank_axis(self):
199205 tm .assert_frame_equal (df .rank (axis = 0 ), df .rank (axis = "index" ))
200206 tm .assert_frame_equal (df .rank (axis = 1 ), df .rank (axis = "columns" ))
201207
208+ @td .skip_if_no_scipy
202209 def test_rank_methods_frame (self ):
203- pytest . importorskip ( " scipy.stats.special" )
204- rankdata = pytest . importorskip ( " scipy.stats. rankdata" )
210+ import scipy .stats # noqa:F401
211+ from scipy .stats import rankdata
205212
206213 xs = np .random .randint (0 , 21 , (100 , 26 ))
207214 xs = (xs - 10.0 ) / 10.0
0 commit comments