test_multilevel.py 文件源码

def test_set_index_period(self):
        # GH 6631
        df = DataFrame(np.random.random(6))
        idx1 = pd.period_range('2011-01-01', periods=3, freq='M')
        idx1 = idx1.append(idx1)
        idx2 = pd.period_range('2013-01-01 09:00', periods=2, freq='H')
        idx2 = idx2.append(idx2).append(idx2)
        idx3 = pd.period_range('2005', periods=6, freq='A')

        df = df.set_index(idx1)
        df = df.set_index(idx2, append=True)
        df = df.set_index(idx3, append=True)

        expected1 = pd.period_range('2011-01-01', periods=3, freq='M')
        expected2 = pd.period_range('2013-01-01 09:00', periods=2, freq='H')

        self.assertTrue(df.index.levels[0].equals(expected1))
        self.assertTrue(df.index.levels[1].equals(expected2))
        self.assertTrue(df.index.levels[2].equals(idx3))

        self.assertTrue(df.index.get_level_values(0).equals(idx1))
        self.assertTrue(df.index.get_level_values(1).equals(idx2))
        self.assertTrue(df.index.get_level_values(2).equals(idx3))