def test_set_index_timestamp():
    df = pd.DataFrame({'A': pd.date_range('2000', periods=12, tz='US/Central'), 'B': 1})
    ddf = dd.from_pandas(df, 2)
    divisions = (pd.Timestamp('2000-01-01 00:00:00-0600', tz='US/Central'), pd.Timestamp('2000-01-12 00:00:00-0600', tz='US/Central'))
    df2 = df.set_index('A')
    ddf_new_div = ddf.set_index('A', divisions=divisions)
    for (ts1, ts2) in zip(divisions, ddf_new_div.divisions):
        assert (ts1.value == ts2.value)
        assert (ts1.tz == ts2.tz)
    assert_eq(df2, ddf_new_div, **CHECK_FREQ)
    assert_eq(df2, ddf.set_index('A'), **CHECK_FREQ)
def test_set_index_timestamp():
    df = pd.DataFrame({'A': pd.date_range('2000', periods=12, tz='US/Central'), 'B': 1})
    ddf = dd.from_pandas(df, 2)
    divisions = (pd.Timestamp('2000-01-01 00:00:00-0600', tz='US/Central'), pd.Timestamp('2000-01-12 00:00:00-0600', tz='US/Central'))
    df2 = df.set_index('A')
    ddf_new_div = ddf.set_index('A', divisions=divisions)
    for (ts1, ts2) in zip(divisions, ddf_new_div.divisions):
        assert (ts1.value == ts2.value)
        assert (ts1.tz == ts2.tz)
    assert_eq(df2, ddf_new_div, **CHECK_FREQ)
    assert_eq(df2, ddf.set_index('A'), **CHECK_FREQ)