def test_cutoff_time_binning(entityset): cutoff_time = pd.DataFrame({ 'time': [ datetime(2011, 4, 9, 12, 31), datetime(2011, 4, 10, 11), datetime(2011, 4, 10, 13, 10, 1) ], 'instance_id': [1, 2, 3] }) binned_cutoff_times = bin_cutoff_times(cutoff_time, Timedelta(4, 'h')) labels = [ datetime(2011, 4, 9, 12), datetime(2011, 4, 10, 8), datetime(2011, 4, 10, 12) ] for i in binned_cutoff_times.index: assert binned_cutoff_times['time'][i] == labels[i] binned_cutoff_times = bin_cutoff_times(cutoff_time, Timedelta(25, 'h')) labels = [ datetime(2011, 4, 8, 22), datetime(2011, 4, 9, 23), datetime(2011, 4, 9, 23) ] for i in binned_cutoff_times.index: assert binned_cutoff_times['time'][i] == labels[i]
def test_cutoff_time_binning(entityset): cutoff_time = pd.DataFrame({'time': [ datetime(2011, 4, 9, 12, 31), datetime(2011, 4, 10, 11), datetime(2011, 4, 10, 13, 10, 1) ], 'instance_id': [1, 2, 3]}) binned_cutoff_times = bin_cutoff_times(cutoff_time, Timedelta(4, 'h')) labels = [datetime(2011, 4, 9, 12), datetime(2011, 4, 10, 8), datetime(2011, 4, 10, 12)] for i in binned_cutoff_times.index: assert binned_cutoff_times['time'][i] == labels[i] binned_cutoff_times = bin_cutoff_times(cutoff_time, Timedelta(25, 'h')) labels = [datetime(2011, 4, 8, 22), datetime(2011, 4, 9, 23), datetime(2011, 4, 9, 23)] for i in binned_cutoff_times.index: assert binned_cutoff_times['time'][i] == labels[i]