Ejemplo n.º 1
0
    def test_concat_periods(self):
        periods = pd.period_range('2000-01-01', periods=10)
        coords = [IndexVariable('t', periods[:5]), IndexVariable('t', periods[5:])]
        expected = IndexVariable('t', periods)
        actual = IndexVariable.concat(coords, dim='t')
        assert actual.identical(expected)
        assert isinstance(actual.to_index(), pd.PeriodIndex)

        positions = [list(range(5)), list(range(5, 10))]
        actual = IndexVariable.concat(coords, dim='t', positions=positions)
        assert actual.identical(expected)
        assert isinstance(actual.to_index(), pd.PeriodIndex)
Ejemplo n.º 2
0
 def test_concat_multiindex(self):
     idx = pd.MultiIndex.from_product([[0, 1, 2], ['a', 'b']])
     coords = [IndexVariable('x', idx[:2]), IndexVariable('x', idx[2:])]
     expected = IndexVariable('x', idx)
     actual = IndexVariable.concat(coords, dim='x')
     assert actual.identical(expected)
     assert isinstance(actual.to_index(), pd.MultiIndex)