Example #1
0
    def test_date_to_index(self):
        dr = date_range('20000101', periods=3000, freq='W')
        y = Series(np.arange(3000.0), index=dr)
        date_index = y.index
        import datetime as dt

        index = date_to_index(date_index[0], date_index)
        assert_equal(index, 0)
        index = date_to_index(date_index[-1], date_index)
        assert_equal(index, date_index.shape[0] - 1)

        index = date_to_index('2009-08-02', date_index)
        assert_equal(index, 500)
        index = date_to_index('2009-08-04', date_index)
        assert_equal(index, 500)
        index = date_to_index('2009-08-01', date_index)
        assert_equal(index, 499)
        index = date_to_index(dt.datetime(2009, 8, 1), date_index)
        assert_equal(index, 499)
        with pytest.raises(ValueError):
            date_to_index(dt.date(2009, 8, 1), date_index)
        z = y + 0.0
        z.index = np.arange(3000)
        num_index = z.index
        with pytest.raises(ValueError):
            date_to_index(dt.datetime(2009, 8, 1), num_index)
Example #2
0
def test_date_to_index_ndarray():
    dr = date_range("20000101", periods=3000, freq="W")
    y = Series(np.arange(3000.0), index=dr)
    date_index = np.asarray(y.index)

    index = date_to_index(date_index[0], date_index)
    assert_equal(index, 0)
    index = date_to_index(date_index[-1], date_index)
    assert_equal(index, date_index.shape[0] - 1)
Example #3
0
    def test_(self):
        dr = date_range('20000101', periods=3000, freq='W')
        y = Series(np.arange(3000.0), index=dr)
        date_index = y.index

        date = date_index[1000] + Timedelta(1, 'D')
        date_pydt = date.to_pydatetime()
        date_npdt = date.to_datetime64()
        date_str = date_pydt.strftime('%Y-%m-%d')
        index = date_to_index(date, date_index)
        index_pydt = date_to_index(date_pydt, date_index)
        index_npdt = date_to_index(date_npdt, date_index)
        index_str = date_to_index(date_str, date_index)
        assert_equal(index, 1001)
        assert_equal(index, index_npdt)
        assert_equal(index, index_pydt)
        assert_equal(index, index_str)

        date = date_index[0] - Timedelta(1, 'D')
        index = date_to_index(date, date_index)
        assert_equal(index, 0)

        date_pydt = date.to_pydatetime()
        date_npdt = date.to_datetime64()
        date_str = date_pydt.strftime('%Y-%m-%d')
        index_pydt = date_to_index(date_pydt, date_index)
        index_npdt = date_to_index(date_npdt, date_index)
        index_str = date_to_index(date_str, date_index)
        assert_equal(index, index_npdt)
        assert_equal(index, index_pydt)
        assert_equal(index, index_str)
Example #4
0
def test_():
    dr = date_range("20000101", periods=3000, freq="W")
    y = Series(np.arange(3000.0), index=dr)
    date_index = y.index

    date = date_index[1000] + Timedelta(1, "D")
    date_pydt = date.to_pydatetime()
    date_npdt = date.to_datetime64()
    date_str = date_pydt.strftime("%Y-%m-%d")
    index = date_to_index(date, date_index)
    index_pydt = date_to_index(date_pydt, date_index)
    index_npdt = date_to_index(date_npdt, date_index)
    index_str = date_to_index(date_str, date_index)
    assert_equal(index, 1001)
    assert_equal(index, index_npdt)
    assert_equal(index, index_pydt)
    assert_equal(index, index_str)

    date = date_index[0] - Timedelta(1, "D")
    index = date_to_index(date, date_index)
    assert_equal(index, 0)

    date_pydt = date.to_pydatetime()
    date_npdt = date.to_datetime64()
    date_str = date_pydt.strftime("%Y-%m-%d")
    index_pydt = date_to_index(date_pydt, date_index)
    index_npdt = date_to_index(date_npdt, date_index)
    index_str = date_to_index(date_str, date_index)
    assert_equal(index, index_npdt)
    assert_equal(index, index_pydt)
    assert_equal(index, index_str)
Example #5
0
    def test_(self):
        dr = date_range('20000101', periods=3000, freq='W')
        y = Series(np.arange(3000.0), index=dr)
        date_index = y.index

        date = date_index[1000] + Timedelta(1, 'D')
        date_pydt = date.to_pydatetime()
        date_npdt = date.to_datetime64()
        date_str = date_pydt.strftime('%Y-%m-%d')
        index = date_to_index(date, date_index)
        index_pydt = date_to_index(date_pydt, date_index)
        index_npdt = date_to_index(date_npdt, date_index)
        index_str = date_to_index(date_str, date_index)
        assert_equal(index, 1001)
        assert_equal(index, index_npdt)
        assert_equal(index, index_pydt)
        assert_equal(index, index_str)

        date = date_index[0] - Timedelta(1, 'D')
        index = date_to_index(date, date_index)
        assert_equal(index, 0)

        date_pydt = date.to_pydatetime()
        date_npdt = date.to_datetime64()
        date_str = date_pydt.strftime('%Y-%m-%d')
        index_pydt = date_to_index(date_pydt, date_index)
        index_npdt = date_to_index(date_npdt, date_index)
        index_str = date_to_index(date_str, date_index)
        assert_equal(index, index_npdt)
        assert_equal(index, index_pydt)
        assert_equal(index, index_str)
Example #6
0
    def test_date_to_index_timestamp(self):
        dr = date_range("20000101", periods=3000, freq="W")
        y = Series(np.arange(3000.0), index=dr)
        date_index = y.index
        date = y.index[1000]
        date_pydt = date.to_pydatetime()
        date_npdt = date.to_datetime64()
        date_str = date_pydt.strftime("%Y-%m-%d")

        index = date_to_index(date, date_index)
        index_pydt = date_to_index(date_pydt, date_index)
        index_npdt = date_to_index(date_npdt, date_index)
        index_str = date_to_index(date_str, date_index)
        assert_equal(index, 1000)
        assert_equal(index, index_npdt)
        assert_equal(index, index_pydt)
        assert_equal(index, index_str)
Example #7
0
    def test_date_to_index(self):
        dr = date_range('20000101', periods=3000, freq='W')
        y = Series(np.arange(3000.0), index=dr)
        date_index = y.index
        import datetime as dt

        index = date_to_index(date_index[0], date_index)
        assert_equal(index, 0)
        index = date_to_index(date_index[-1], date_index)
        assert_equal(index, date_index.shape[0] - 1)

        index = date_to_index('2009-08-02', date_index)
        assert_equal(index, 500)
        index = date_to_index('2009-08-04', date_index)
        assert_equal(index, 500)
        index = date_to_index('2009-08-01', date_index)
        assert_equal(index, 499)
        index = date_to_index(dt.datetime(2009, 8, 1), date_index)
        assert_equal(index, 499)
        assert_raises(ValueError, date_to_index, dt.date(2009, 8, 1),
                      date_index)
        z = y + 0.0
        z.index = np.arange(3000)
        num_index = z.index
        assert_raises(ValueError, date_to_index,
                      dt.datetime(2009, 8, 1), num_index)
Example #8
0
    def test_date_to_index_timestamp(self):
        dr = date_range('20000101', periods=3000, freq='W')
        y = Series(np.arange(3000.0), index=dr)
        date_index = y.index
        date = y.index[1000]

        try:
            date_pydt = date.to_pydatetime()
        except AttributeError:
            # Old pandas
            date_pydt = date.to_datetime()

        date_npdt = date.to_datetime64()
        date_str = date_pydt.strftime('%Y-%m-%d')
        index = date_to_index(date, date_index)
        index_pydt = date_to_index(date_pydt, date_index)
        index_npdt = date_to_index(date_npdt, date_index)
        index_str = date_to_index(date_str, date_index)
        assert_equal(index, 1000)
        assert_equal(index, index_npdt)
        assert_equal(index, index_pydt)
        assert_equal(index, index_str)
Example #9
0
    def test_date_to_index(self):
        dr = date_range('20000101', periods=3000, freq='W')
        y = Series(np.arange(3000.0), index=dr)
        date_index = y.index

        index = date_to_index(date_index[0], date_index)
        assert_equal(index, 0)
        index = date_to_index(date_index[-1], date_index)
        assert_equal(index, date_index.shape[0] - 1)

        index = date_to_index('2009-08-02', date_index)
        assert_equal(index, 500)
        index = date_to_index('2009-08-04', date_index)
        assert_equal(index, 501)
        index = date_to_index('2009-08-01', date_index)
        assert_equal(index, 500)
        index = date_to_index(dt.datetime(2009, 8, 1), date_index)
        assert_equal(index, 500)
        with pytest.raises(ValueError):
            date_to_index(dt.date(2009, 8, 1), date_index)
        z = y + 0.0
        z.index = np.arange(3000)
        num_index = z.index
        with pytest.raises(ValueError):
            date_to_index(dt.datetime(2009, 8, 1), num_index)
        idx = date_range('1999-12-31', periods=3)

        df = DataFrame([1, 2, 3], index=idx[::-1])
        with pytest.raises(ValueError):
            date_to_index(idx[0], df.index)

        df = DataFrame([1, 2, 3], index=[idx[0]] * 3)
        with pytest.raises(ValueError):
            date_to_index(idx[0], df.index)

        with pytest.raises(ValueError):
            date_to_index('NaT', idx)

        #  check whether this also works for a localized datetimeindex
        date_index = date_range('20000101',
                                periods=3000,
                                freq='W',
                                tz='Europe/Berlin')
        index = date_to_index(date_index[0], date_index)
        assert_equal(index, 0)
Example #10
0
    def test_date_to_index(self):
        dr = date_range('20000101', periods=3000, freq='W')
        y = Series(np.arange(3000.0), index=dr)
        date_index = y.index

        index = date_to_index(date_index[0], date_index)
        assert_equal(index, 0)
        index = date_to_index(date_index[-1], date_index)
        assert_equal(index, date_index.shape[0] - 1)

        index = date_to_index('2009-08-02', date_index)
        assert_equal(index, 500)
        index = date_to_index('2009-08-04', date_index)
        assert_equal(index, 501)
        index = date_to_index('2009-08-01', date_index)
        assert_equal(index, 500)
        index = date_to_index(dt.datetime(2009, 8, 1), date_index)
        assert_equal(index, 500)
        with pytest.raises(ValueError):
            date_to_index(dt.date(2009, 8, 1), date_index)
        z = y + 0.0
        z.index = np.arange(3000)
        num_index = z.index
        with pytest.raises(ValueError):
            date_to_index(dt.datetime(2009, 8, 1), num_index)
        idx = date_range('1999-12-31', periods=3)

        df = DataFrame([1, 2, 3], index=idx[::-1])
        with pytest.raises(ValueError):
            date_to_index(idx[0], df.index)

        df = DataFrame([1, 2, 3], index=[idx[0]] * 3)
        with pytest.raises(ValueError):
            date_to_index(idx[0], df.index)

        with pytest.raises(ValueError):
            date_to_index('NaT', idx)

        #  check whether this also works for a localized datetimeindex
        date_index = date_range('20000101', periods=3000, freq='W', tz='Europe/Berlin')
        index = date_to_index(date_index[0], date_index)
        assert_equal(index, 0)