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)
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)
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)
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)
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)
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)
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)
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)