/usr/local/lib64/python3.6/site-packages/pandas/tests/groupby
NameSizeModeActions
aggregate/-0755rm
transform/-0755rm
__pycache__/-0755rm
conftest.py34250644editdlrm
test_allowlist.py104100644editdlrm
test_apply.py317000644editdlrm
test_apply_mutate.py18490644editdlrm
test_bin_groupby.py40830644editdlrm
test_categorical.py561370644editdlrm
test_counting.py122890644editdlrm
test_filters.py204020644editdlrm
test_function.py329130644editdlrm
test_groupby.py625940644editdlrm
test_groupby_dropna.py71930644editdlrm
test_groupby_subclass.py26760644editdlrm
test_grouping.py350910644editdlrm
test_index_as_string.py20690644editdlrm
test_nth.py211120644editdlrm
test_nunique.py58030644editdlrm
test_pipe.py20750644editdlrm
test_quantile.py82360644editdlrm
test_rank.py154470644editdlrm
test_sample.py43580644editdlrm
test_size.py21490644editdlrm
test_timegrouper.py284530644editdlrm
test_value_counts.py34700644editdlrm
__init__.py00644editdlrm
Edit: /usr/local/lib64/python3.6/site-packages/pandas/tests/groupby/conftest.py (3425B)
import numpy as np import pytest from pandas import DataFrame, MultiIndex import pandas._testing as tm from pandas.core.groupby.base import reduction_kernels, transformation_kernels @pytest.fixture def mframe(): index = MultiIndex( levels=[["foo", "bar", "baz", "qux"], ["one", "two", "three"]], codes=[[0, 0, 0, 1, 1, 2, 2, 3, 3, 3], [0, 1, 2, 0, 1, 1, 2, 0, 1, 2]], names=["first", "second"], ) return DataFrame(np.random.randn(10, 3), index=index, columns=["A", "B", "C"]) @pytest.fixture def df(): return DataFrame( { "A": ["foo", "bar", "foo", "bar", "foo", "bar", "foo", "foo"], "B": ["one", "one", "two", "three", "two", "two", "one", "three"], "C": np.random.randn(8), "D": np.random.randn(8), } ) @pytest.fixture def ts(): return tm.makeTimeSeries() @pytest.fixture def tsd(): return tm.getTimeSeriesData() @pytest.fixture def tsframe(tsd): return DataFrame(tsd) @pytest.fixture def df_mixed_floats(): return DataFrame( { "A": ["foo", "bar", "foo", "bar", "foo", "bar", "foo", "foo"], "B": ["one", "one", "two", "three", "two", "two", "one", "three"], "C": np.random.randn(8), "D": np.array(np.random.randn(8), dtype="float32"), } ) @pytest.fixture def three_group(): return DataFrame( { "A": [ "foo", "foo", "foo", "foo", "bar", "bar", "bar", "bar", "foo", "foo", "foo", ], "B": [ "one", "one", "one", "two", "one", "one", "one", "two", "two", "two", "one", ], "C": [ "dull", "dull", "shiny", "dull", "dull", "shiny", "shiny", "dull", "shiny", "shiny", "shiny", ], "D": np.random.randn(11), "E": np.random.randn(11), "F": np.random.randn(11), } ) @pytest.fixture(params=sorted(reduction_kernels)) def reduction_func(request): """ yields the string names of all groupby reduction functions, one at a time. """ return request.param @pytest.fixture(params=sorted(transformation_kernels)) def transformation_func(request): """yields the string names of all groupby transformation functions.""" return request.param @pytest.fixture(params=sorted(reduction_kernels) + sorted(transformation_kernels)) def groupby_func(request): """yields both aggregation and transformation functions.""" return request.param @pytest.fixture(params=[True, False]) def parallel(request): """parallel keyword argument for numba.jit""" return request.param @pytest.fixture(params=[True, False]) def nogil(request): """nogil keyword argument for numba.jit""" return request.param @pytest.fixture(params=[True, False]) def nopython(request): """nopython keyword argument for numba.jit""" return request.param