Skip to content

Commit e48d1e4

Browse files
simonjayhawkinsPingviinituutti
simonjayhawkins
authored and
Pingviinituutti
committed
TST: isort tests/groupby (pandas-dev#24553)
1 parent 6dfa838 commit e48d1e4

20 files changed

+108
-117
lines changed

‎pandas/tests/groupby/aggregate/test_aggregate.py

+5-5
Original file line numberDiff line numberDiff line change
@@ -4,15 +4,15 @@
44
test .agg behavior / note that .apply is tested generally in test_groupby.py
55
"""
66

7+
import numpy as np
78
import pytest
89

9-
import numpy as np
10-
import pandas as pd
10+
from pandas.compat import OrderedDict
1111

12-
from pandas import concat, DataFrame, Index, MultiIndex, Series
13-
from pandas.core.groupby.grouper import Grouping
12+
import pandas as pd
13+
from pandas import DataFrame, Index, MultiIndex, Series, concat
1414
from pandas.core.base import SpecificationError
15-
from pandas.compat import OrderedDict
15+
from pandas.core.groupby.grouper import Grouping
1616
import pandas.util.testing as tm
1717

1818

‎pandas/tests/groupby/aggregate/test_cython.py

+3-4
Original file line numberDiff line numberDiff line change
@@ -6,13 +6,12 @@
66

77
from __future__ import print_function
88

9+
import numpy as np
910
import pytest
1011

11-
import numpy as np
1212
import pandas as pd
13-
14-
from pandas import (bdate_range, DataFrame, Index, Series, Timestamp,
15-
Timedelta, NaT)
13+
from pandas import (
14+
DataFrame, Index, NaT, Series, Timedelta, Timestamp, bdate_range)
1615
from pandas.core.groupby.groupby import DataError
1716
import pandas.util.testing as tm
1817

‎pandas/tests/groupby/aggregate/test_other.py

+6-6
Original file line numberDiff line numberDiff line change
@@ -6,22 +6,22 @@
66

77
from __future__ import print_function
88

9-
import pytest
109
from collections import OrderedDict
11-
1210
import datetime as dt
1311
from functools import partial
1412

1513
import numpy as np
16-
import pandas as pd
14+
import pytest
1715

16+
import pandas as pd
1817
from pandas import (
19-
date_range, DataFrame, Index, MultiIndex, PeriodIndex, period_range, Series
20-
)
18+
DataFrame, Index, MultiIndex, PeriodIndex, Series, date_range,
19+
period_range)
2120
from pandas.core.groupby.groupby import SpecificationError
22-
from pandas.io.formats.printing import pprint_thing
2321
import pandas.util.testing as tm
2422

23+
from pandas.io.formats.printing import pprint_thing
24+
2525

2626
def test_agg_api():
2727
# GH 6337

‎pandas/tests/groupby/conftest.py

+3-2
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
1-
import pytest
21
import numpy as np
3-
from pandas import MultiIndex, DataFrame
2+
import pytest
3+
4+
from pandas import DataFrame, MultiIndex
45
from pandas.util import testing as tm
56

67

‎pandas/tests/groupby/test_apply.py

+5-3
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,11 @@
1-
import pytest
1+
from datetime import datetime
2+
23
import numpy as np
4+
import pytest
5+
36
import pandas as pd
4-
from datetime import datetime
7+
from pandas import DataFrame, Index, MultiIndex, Series, bdate_range, compat
58
from pandas.util import testing as tm
6-
from pandas import DataFrame, MultiIndex, compat, Series, bdate_range, Index
79

810

911
def test_apply_issues():

‎pandas/tests/groupby/test_bin_groupby.py

+5-4
Original file line numberDiff line numberDiff line change
@@ -1,16 +1,17 @@
11
# -*- coding: utf-8 -*-
22

3+
import numpy as np
4+
from numpy import nan
35
import pytest
46

5-
from numpy import nan
6-
import numpy as np
7+
from pandas._libs import groupby, lib, reduction
78

89
from pandas.core.dtypes.common import ensure_int64
10+
911
from pandas import Index, isna
1012
from pandas.core.groupby.ops import generate_bins_generic
11-
from pandas.util.testing import assert_almost_equal
1213
import pandas.util.testing as tm
13-
from pandas._libs import lib, groupby, reduction
14+
from pandas.util.testing import assert_almost_equal
1415

1516

1617
def test_series_grouper():

‎pandas/tests/groupby/test_categorical.py

+8-6
Original file line numberDiff line numberDiff line change
@@ -1,17 +1,19 @@
11
# -*- coding: utf-8 -*-
22
from __future__ import print_function
3+
34
from datetime import datetime
45

6+
import numpy as np
57
import pytest
68

7-
import numpy as np
8-
import pandas as pd
99
from pandas.compat import PY37
10-
from pandas import (Index, MultiIndex, CategoricalIndex,
11-
DataFrame, Categorical, Series, qcut)
12-
from pandas.util.testing import (assert_equal,
13-
assert_frame_equal, assert_series_equal)
10+
11+
import pandas as pd
12+
from pandas import (
13+
Categorical, CategoricalIndex, DataFrame, Index, MultiIndex, Series, qcut)
1414
import pandas.util.testing as tm
15+
from pandas.util.testing import (
16+
assert_equal, assert_frame_equal, assert_series_equal)
1517

1618

1719
def cartesian_product_for_groupers(result, args, names):

‎pandas/tests/groupby/test_counting.py

+4-4
Original file line numberDiff line numberDiff line change
@@ -4,10 +4,10 @@
44
import numpy as np
55
import pytest
66

7-
from pandas import (DataFrame, Series, MultiIndex, Timestamp, Timedelta,
8-
Period)
9-
from pandas.util.testing import (assert_series_equal, assert_frame_equal)
10-
from pandas.compat import (range, product as cart_product)
7+
from pandas.compat import product as cart_product, range
8+
9+
from pandas import DataFrame, MultiIndex, Period, Series, Timedelta, Timestamp
10+
from pandas.util.testing import assert_frame_equal, assert_series_equal
1111

1212

1313
class TestCounting(object):

‎pandas/tests/groupby/test_filters.py

+4-3
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,12 @@
11
# -*- coding: utf-8 -*-
22
from __future__ import print_function
3-
import pytest
43

54
import numpy as np
6-
import pandas.util.testing as tm
7-
from pandas import Timestamp, DataFrame, Series
5+
import pytest
6+
87
import pandas as pd
8+
from pandas import DataFrame, Series, Timestamp
9+
import pandas.util.testing as tm
910

1011

1112
def test_filter_series():

‎pandas/tests/groupby/test_function.py

+9-7
Original file line numberDiff line numberDiff line change
@@ -1,14 +1,16 @@
1-
import pytest
1+
from string import ascii_lowercase
22

33
import numpy as np
4-
import pandas as pd
5-
from pandas import (DataFrame, Index, compat, isna,
6-
Series, MultiIndex, Timestamp, date_range)
4+
import pytest
5+
6+
from pandas.compat import product as cart_product
77
from pandas.errors import UnsupportedFunctionCall
8-
from pandas.util import testing as tm
8+
9+
import pandas as pd
10+
from pandas import (
11+
DataFrame, Index, MultiIndex, Series, Timestamp, compat, date_range, isna)
912
import pandas.core.nanops as nanops
10-
from string import ascii_lowercase
11-
from pandas.compat import product as cart_product
13+
from pandas.util import testing as tm
1214

1315

1416
@pytest.mark.parametrize("agg_func", ['any', 'all'])

‎pandas/tests/groupby/test_groupby.py

+13-14
Original file line numberDiff line numberDiff line change
@@ -1,26 +1,25 @@
11
# -*- coding: utf-8 -*-
22
from __future__ import print_function
33

4-
import pytest
5-
4+
from collections import defaultdict
65
from datetime import datetime
76
from decimal import Decimal
87

9-
from pandas import (date_range, Timestamp,
10-
Index, MultiIndex, DataFrame, Series,
11-
Panel, read_csv)
12-
from pandas.errors import PerformanceWarning
13-
from pandas.util.testing import (assert_frame_equal,
14-
assert_series_equal, assert_almost_equal)
15-
from pandas.compat import (range, lrange, StringIO, lmap, lzip, map, zip,
16-
OrderedDict)
17-
from pandas import compat
18-
from collections import defaultdict
19-
import pandas.core.common as com
208
import numpy as np
9+
import pytest
10+
11+
from pandas.compat import (
12+
OrderedDict, StringIO, lmap, lrange, lzip, map, range, zip)
13+
from pandas.errors import PerformanceWarning
2114

22-
import pandas.util.testing as tm
2315
import pandas as pd
16+
from pandas import (
17+
DataFrame, Index, MultiIndex, Panel, Series, Timestamp, compat, date_range,
18+
read_csv)
19+
import pandas.core.common as com
20+
import pandas.util.testing as tm
21+
from pandas.util.testing import (
22+
assert_almost_equal, assert_frame_equal, assert_series_equal)
2423

2524

2625
def test_repr():

‎pandas/tests/groupby/test_grouping.py

+11-11
Original file line numberDiff line numberDiff line change
@@ -2,25 +2,25 @@
22

33
""" test where we are determining what we are grouping, or getting groups """
44

5+
import numpy as np
56
import pytest
67

7-
from pandas import (date_range, Timestamp,
8-
Index, MultiIndex, DataFrame, Series, CategoricalIndex)
9-
from pandas.util.testing import (assert_panel_equal, assert_frame_equal,
10-
assert_series_equal, assert_almost_equal)
11-
from pandas.core.groupby.grouper import Grouping
12-
from pandas.compat import lrange, long
8+
from pandas.compat import long, lrange
139

14-
from pandas import compat
15-
import numpy as np
16-
17-
import pandas.util.testing as tm
1810
import pandas as pd
19-
11+
from pandas import (
12+
CategoricalIndex, DataFrame, Index, MultiIndex, Series, Timestamp, compat,
13+
date_range)
14+
from pandas.core.groupby.grouper import Grouping
15+
import pandas.util.testing as tm
16+
from pandas.util.testing import (
17+
assert_almost_equal, assert_frame_equal, assert_panel_equal,
18+
assert_series_equal)
2019

2120
# selection
2221
# --------------------------------
2322

23+
2424
class TestSelection(object):
2525

2626
def test_select_bad_cols(self):

‎pandas/tests/groupby/test_index_as_string.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
1-
import pytest
2-
import pandas as pd
31
import numpy as np
2+
import pytest
43

4+
import pandas as pd
55
from pandas.util.testing import assert_frame_equal, assert_series_equal
66

77

‎pandas/tests/groupby/test_nth.py

+6-6
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,12 @@
11
import numpy as np
2-
import pandas as pd
3-
from pandas import DataFrame, MultiIndex, Index, Series, isna, Timestamp
2+
import pytest
3+
44
from pandas.compat import lrange
5+
6+
import pandas as pd
7+
from pandas import DataFrame, Index, MultiIndex, Series, Timestamp, isna
58
from pandas.util.testing import (
6-
assert_frame_equal,
7-
assert_produces_warning,
8-
assert_series_equal)
9-
import pytest
9+
assert_frame_equal, assert_produces_warning, assert_series_equal)
1010

1111

1212
def test_first_last_nth(df):

‎pandas/tests/groupby/test_rank.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
1-
import pytest
21
import numpy as np
2+
import pytest
3+
34
import pandas as pd
45
from pandas import DataFrame, Series, concat
56
from pandas.util import testing as tm

‎pandas/tests/groupby/test_timegrouper.py

+6-6
Original file line numberDiff line numberDiff line change
@@ -1,17 +1,17 @@
11
""" test with the TimeGrouper / grouping with datetimes """
22

3-
import pytest
4-
import pytz
5-
63
from datetime import datetime
4+
75
import numpy as np
86
from numpy import nan
7+
import pytest
8+
import pytz
9+
10+
from pandas.compat import StringIO
911

1012
import pandas as pd
11-
from pandas import (DataFrame, date_range, Index,
12-
Series, MultiIndex, Timestamp)
13+
from pandas import DataFrame, Index, MultiIndex, Series, Timestamp, date_range
1314
from pandas.core.groupby.ops import BinGrouper
14-
from pandas.compat import StringIO
1515
from pandas.util import testing as tm
1616
from pandas.util.testing import assert_frame_equal, assert_series_equal
1717

‎pandas/tests/groupby/test_transform.py

+9-9
Original file line numberDiff line numberDiff line change
@@ -1,19 +1,19 @@
11
""" test with the .transform """
22

3+
import numpy as np
34
import pytest
45

5-
import numpy as np
6-
import pandas as pd
7-
from pandas.util import testing as tm
8-
from pandas import Series, DataFrame, Timestamp, MultiIndex, concat, date_range
9-
from pandas.core.dtypes.common import (
10-
ensure_platform_int, is_timedelta64_dtype)
11-
from pandas.compat import StringIO
126
from pandas._libs import groupby
7+
from pandas.compat import StringIO
138

14-
from pandas.util.testing import assert_frame_equal, assert_series_equal
15-
from pandas.core.groupby.groupby import DataError
9+
from pandas.core.dtypes.common import ensure_platform_int, is_timedelta64_dtype
10+
11+
import pandas as pd
12+
from pandas import DataFrame, MultiIndex, Series, Timestamp, concat, date_range
1613
from pandas.core.config import option_context
14+
from pandas.core.groupby.groupby import DataError
15+
from pandas.util import testing as tm
16+
from pandas.util.testing import assert_frame_equal, assert_series_equal
1717

1818

1919
def assert_fp_equal(a, b):

‎pandas/tests/groupby/test_value_counts.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -4,13 +4,13 @@
44
and proper parameter handling
55
"""
66

7-
import pytest
8-
97
from itertools import product
8+
109
import numpy as np
10+
import pytest
1111

12+
from pandas import DataFrame, MultiIndex, Series, date_range
1213
from pandas.util import testing as tm
13-
from pandas import MultiIndex, DataFrame, Series, date_range
1414

1515

1616
# our starting frame

‎pandas/tests/groupby/test_whitelist.py

+4-2
Original file line numberDiff line numberDiff line change
@@ -3,10 +3,12 @@
33
the so-called white/black lists
44
"""
55

6-
import pytest
76
from string import ascii_lowercase
7+
88
import numpy as np
9-
from pandas import DataFrame, Series, compat, date_range, Index, MultiIndex
9+
import pytest
10+
11+
from pandas import DataFrame, Index, MultiIndex, Series, compat, date_range
1012
from pandas.util import testing as tm
1113

1214
AGG_FUNCTIONS = ['sum', 'prod', 'min', 'max', 'median', 'mean', 'skew',

0 commit comments

Comments
 (0)