Exemple #1
0
    def test_resample(self):
        ii1 = IntervalIndex(freq='A', start='1/1/2001', end='1/1/2001')
        ii2 = IntervalIndex(freq='Q', start='1/1/2001', end='1/1/2001')
        ii3 = IntervalIndex(freq='M', start='1/1/2001', end='1/1/2001')
        ii4 = IntervalIndex(freq='D', start='1/1/2001', end='1/1/2001')
        ii5 = IntervalIndex(freq='H', start='1/1/2001', end='1/1/2001 00:00')
        ii6 = IntervalIndex(freq='Min', start='1/1/2001', end='1/1/2001 00:00')
        ii7 = IntervalIndex(freq='S', start='1/1/2001', end='1/1/2001 00:00:00')

        self.assertEquals(ii1.resample('Q', 'S'), ii2)
        self.assertEquals(ii1.resample('Q', 's'), ii2)
        self.assertEquals(ii1.resample('M', 'start'), ii3)
        self.assertEquals(ii1.resample('D', 'StarT'), ii4)
        self.assertEquals(ii1.resample('H', 'beGIN'), ii5)
        self.assertEquals(ii1.resample('Min', 'S'), ii6)
        self.assertEquals(ii1.resample('S', 'S'), ii7)

        self.assertEquals(ii2.resample('A', 'S'), ii1)
        self.assertEquals(ii2.resample('M', 'S'), ii3)
        self.assertEquals(ii2.resample('D', 'S'), ii4)
        self.assertEquals(ii2.resample('H', 'S'), ii5)
        self.assertEquals(ii2.resample('Min', 'S'), ii6)
        self.assertEquals(ii2.resample('S', 'S'), ii7)

        self.assertEquals(ii3.resample('A', 'S'), ii1)
        self.assertEquals(ii3.resample('Q', 'S'), ii2)
        self.assertEquals(ii3.resample('D', 'S'), ii4)
        self.assertEquals(ii3.resample('H', 'S'), ii5)
        self.assertEquals(ii3.resample('Min', 'S'), ii6)
        self.assertEquals(ii3.resample('S', 'S'), ii7)

        self.assertEquals(ii4.resample('A', 'S'), ii1)
        self.assertEquals(ii4.resample('Q', 'S'), ii2)
        self.assertEquals(ii4.resample('M', 'S'), ii3)
        self.assertEquals(ii4.resample('H', 'S'), ii5)
        self.assertEquals(ii4.resample('Min', 'S'), ii6)
        self.assertEquals(ii4.resample('S', 'S'), ii7)

        self.assertEquals(ii5.resample('A', 'S'), ii1)
        self.assertEquals(ii5.resample('Q', 'S'), ii2)
        self.assertEquals(ii5.resample('M', 'S'), ii3)
        self.assertEquals(ii5.resample('D', 'S'), ii4)
        self.assertEquals(ii5.resample('Min', 'S'), ii6)
        self.assertEquals(ii5.resample('S', 'S'), ii7)

        self.assertEquals(ii6.resample('A', 'S'), ii1)
        self.assertEquals(ii6.resample('Q', 'S'), ii2)
        self.assertEquals(ii6.resample('M', 'S'), ii3)
        self.assertEquals(ii6.resample('D', 'S'), ii4)
        self.assertEquals(ii6.resample('H', 'S'), ii5)
        self.assertEquals(ii6.resample('S', 'S'), ii7)

        self.assertEquals(ii7.resample('A', 'S'), ii1)
        self.assertEquals(ii7.resample('Q', 'S'), ii2)
        self.assertEquals(ii7.resample('M', 'S'), ii3)
        self.assertEquals(ii7.resample('D', 'S'), ii4)
        self.assertEquals(ii7.resample('H', 'S'), ii5)
        self.assertEquals(ii7.resample('Min', 'S'), ii6)
Exemple #2
0
    def test_shift(self):
        ii1 = IntervalIndex(freq='A', start='1/1/2001', end='12/1/2009')
        ii2 = IntervalIndex(freq='A', start='1/1/2002', end='12/1/2010')
        assert_equal(len(ii1), len(ii2))
        assert_equal(ii1.shift(1).values, ii2.values)

        ii1 = IntervalIndex(freq='A', start='1/1/2001', end='12/1/2009')
        ii2 = IntervalIndex(freq='A', start='1/1/2000', end='12/1/2008')
        assert_equal(len(ii1), len(ii2))
        assert_equal(ii1.shift(-1).values, ii2.values)

        ii1 = IntervalIndex(freq='M', start='1/1/2001', end='12/1/2009')
        ii2 = IntervalIndex(freq='M', start='2/1/2001', end='1/1/2010')
        assert_equal(len(ii1), len(ii2))
        assert_equal(ii1.shift(1).values, ii2.values)

        ii1 = IntervalIndex(freq='M', start='1/1/2001', end='12/1/2009')
        ii2 = IntervalIndex(freq='M', start='12/1/2000', end='11/1/2009')
        assert_equal(len(ii1), len(ii2))
        assert_equal(ii1.shift(-1).values, ii2.values)

        ii1 = IntervalIndex(freq='D', start='1/1/2001', end='12/1/2009')
        ii2 = IntervalIndex(freq='D', start='1/2/2001', end='12/2/2009')
        assert_equal(len(ii1), len(ii2))
        assert_equal(ii1.shift(1).values, ii2.values)

        ii1 = IntervalIndex(freq='D', start='1/1/2001', end='12/1/2009')
        ii2 = IntervalIndex(freq='D', start='12/31/2000', end='11/30/2009')
        assert_equal(len(ii1), len(ii2))
        assert_equal(ii1.shift(-1).values, ii2.values)