def test_select_window_end_ts_and_window_size_within_data(self): """Should return dataframe with complete window""" selected_activities = select_activity_window(self.test_activities, self.test_activities.start_date_local[2], 2) self.assertEqual(2, len(selected_activities)) self.assertEqual(2, selected_activities.test_var.values[0]) self.assertEqual(3.5, selected_activities.test_var.values[1])
def test_select_window_end_ts_before_last_activity_window_size_outside_data(self): """Should return first activity""" selected_activities = select_activity_window(self.test_activities, pd.tslib.Timestamp('2015-04-29'), 2) self.assertEqual(1, len(selected_activities)) self.assertEqual(1, selected_activities.test_var.values[0])
def test_select_window_end_ts_after_last_activity_window_size_outside_data(self): """Should return empty""" selected_activities = select_activity_window(self.test_activities, pd.tslib.Timestamp('2015-05-05'), 2) self.assertEqual(0, len(selected_activities))
def test_select_window_end_ts_after_last_activity_window_size_within_data(self): """Should return last activity""" selected_activities = select_activity_window(self.test_activities, pd.tslib.Timestamp('2015-05-02'), 2) self.assertEqual(1, len(selected_activities)) self.assertEqual(3.5, selected_activities.test_var.values[0])