FM / period

/usr/local/lib64/python3.6/site-packages/pandas/tests/indexes/period/

/usr/local/lib64/python3.6/site-packages/pandas/tests/indexes/period

Name	Size	Mode	Actions
__pycache__/	-	0755	rm
test_asfreq.py	5432	0644	edit dl rm
test_astype.py	6189	0644	edit dl rm
test_constructors.py	19831	0644	edit dl rm
test_factorize.py	1267	0644	edit dl rm
test_fillna.py	1104	0644	edit dl rm
test_formats.py	6971	0644	edit dl rm
test_indexing.py	29017	0644	edit dl rm
test_join.py	1410	0644	edit dl rm
test_monotonic.py	1245	0644	edit dl rm
test_ops.py	12642	0644	edit dl rm
test_partial_slicing.py	5323	0644	edit dl rm
test_period.py	21462	0644	edit dl rm
test_period_range.py	3954	0644	edit dl rm
test_scalar_compat.py	1123	0644	edit dl rm
test_searchsorted.py	2622	0644	edit dl rm
test_setops.py	12087	0644	edit dl rm
test_shift.py	4398	0644	edit dl rm
test_tools.py	1006	0644	edit dl rm
test_to_timestamp.py	3612	0644	edit dl rm
__init__.py	0	0644	edit dl rm

Edit: /usr/local/lib64/python3.6/site-packages/pandas/tests/indexes/period/test_factorize.py (1267B)

import numpy as np

from pandas import PeriodIndex
import pandas._testing as tm

class TestFactorize:
    def test_factorize(self):
        idx1 = PeriodIndex(
            ["2014-01", "2014-01", "2014-02", "2014-02", "2014-03", "2014-03"], freq="M"
        )

exp_arr = np.array([0, 0, 1, 1, 2, 2], dtype=np.intp)
        exp_idx = PeriodIndex(["2014-01", "2014-02", "2014-03"], freq="M")

arr, idx = idx1.factorize()
        tm.assert_numpy_array_equal(arr, exp_arr)
        tm.assert_index_equal(idx, exp_idx)

arr, idx = idx1.factorize(sort=True)
        tm.assert_numpy_array_equal(arr, exp_arr)
        tm.assert_index_equal(idx, exp_idx)

idx2 = PeriodIndex(
            ["2014-03", "2014-03", "2014-02", "2014-01", "2014-03", "2014-01"], freq="M"
        )

exp_arr = np.array([2, 2, 1, 0, 2, 0], dtype=np.intp)
        arr, idx = idx2.factorize(sort=True)
        tm.assert_numpy_array_equal(arr, exp_arr)
        tm.assert_index_equal(idx, exp_idx)

exp_arr = np.array([0, 0, 1, 2, 0, 2], dtype=np.intp)
        exp_idx = PeriodIndex(["2014-03", "2014-02", "2014-01"], freq="M")
        arr, idx = idx2.factorize()
        tm.assert_numpy_array_equal(arr, exp_arr)
        tm.assert_index_equal(idx, exp_idx)

import numpy as np from pandas import PeriodIndex import pandas._testing as tm class TestFactorize: def test_factorize(self): idx1 = PeriodIndex( ["2014-01", "2014-01", "2014-02", "2014-02", "2014-03", "2014-03"], freq="M" ) exp_arr = np.array([0, 0, 1, 1, 2, 2], dtype=np.intp) exp_idx = PeriodIndex(["2014-01", "2014-02", "2014-03"], freq="M") arr, idx = idx1.factorize() tm.assert_numpy_array_equal(arr, exp_arr) tm.assert_index_equal(idx, exp_idx) arr, idx = idx1.factorize(sort=True) tm.assert_numpy_array_equal(arr, exp_arr) tm.assert_index_equal(idx, exp_idx) idx2 = PeriodIndex( ["2014-03", "2014-03", "2014-02", "2014-01", "2014-03", "2014-01"], freq="M" ) exp_arr = np.array([2, 2, 1, 0, 2, 0], dtype=np.intp) arr, idx = idx2.factorize(sort=True) tm.assert_numpy_array_equal(arr, exp_arr) tm.assert_index_equal(idx, exp_idx) exp_arr = np.array([0, 0, 1, 2, 0, 2], dtype=np.intp) exp_idx = PeriodIndex(["2014-03", "2014-02", "2014-01"], freq="M") arr, idx = idx2.factorize() tm.assert_numpy_array_equal(arr, exp_arr) tm.assert_index_equal(idx, exp_idx)