def test_timedelta_breaks(): breaks = timedelta_breaks() # cpython x = [timedelta(days=i * 365) for i in range(25)] limits = min(x), max(x) major = breaks(limits) years = [val.total_seconds() / (365 * 24 * 60 * 60) for val in major] npt.assert_array_equal(years, [0, 5, 10, 15, 20, 25]) x = [timedelta(microseconds=i) for i in range(25)] limits = min(x), max(x) major = breaks(limits) mseconds = [val.total_seconds() * 10**6 for val in major] npt.assert_array_equal(mseconds, [0, 5, 10, 15, 20, 25]) # pandas x = [pd.Timedelta(seconds=i * 60) for i in range(10)] limits = min(x), max(x) major = breaks(limits) minutes = [val.total_seconds() / 60 for val in major] npt.assert_allclose(minutes, [0, 2, 4, 6, 8]) # numpy x = [np.timedelta64(i * 10, unit='D') for i in range(1, 10)] limits = min(x), max(x) with pytest.raises(ValueError): breaks(limits) # NaT limits = pd.NaT, pd.Timedelta(seconds=9 * 60) assert len(breaks(limits)) == 0
def test_timedelta_breaks(): breaks = timedelta_breaks() # cpython x = [timedelta(days=i*365) for i in range(25)] limits = min(x), max(x) major = breaks(limits) years = [val.total_seconds()/(365*24*60*60)for val in major] npt.assert_array_equal( years, [0, 5, 10, 15, 20, 25]) x = [timedelta(microseconds=i) for i in range(25)] limits = min(x), max(x) major = breaks(limits) mseconds = [val.total_seconds()*10**6 for val in major] npt.assert_array_equal( mseconds, [0, 5, 10, 15, 20, 25]) # pandas x = [pd.Timedelta(seconds=i*60) for i in range(10)] limits = min(x), max(x) major = breaks(limits) minutes = [val.total_seconds()/60 for val in major] npt.assert_allclose( minutes, [0, 2, 4, 6, 8]) # numpy x = [np.timedelta64(i*10, unit='D') for i in range(1, 10)] limits = min(x), max(x) with pytest.raises(ValueError): breaks(limits) # NaT limits = pd.NaT, pd.Timedelta(seconds=9*60) assert len(breaks(limits)) == 0