def test_start_date_pick_none_of_two_by_undershooting_date(self): start_time = datetime.strptime('2010/04/01 00:00:00', '%Y/%m/%d %H:%M:%S') end_time = datetime.strptime('2010/05/01 00:13:00', '%Y/%m/%d %H:%M:%S') result = filters.filter_on_start_and_end_date(self.start_date_data, start_time=start_time, end_time=end_time)\ .reset_index(drop=True) aim = pd.DataFrame({'START_DATE': [], 'END_DATE': []}) aim['START_DATE'] = pd.to_datetime(aim['START_DATE'], format='%Y/%m/%d %H:%M:%S') aim['END_DATE'] = pd.to_datetime(aim['END_DATE'], format='%Y/%m/%d %H:%M:%S') assert_frame_equal(aim, result)
def test_start_date_pick_first_of_two(self): start_time = datetime.strptime('2014/06/01 00:00:00', '%Y/%m/%d %H:%M:%S') end_time = datetime.strptime('2014/09/01 00:13:00', '%Y/%m/%d %H:%M:%S') result = filters.filter_on_start_and_end_date(self.start_date_data, start_time=start_time, end_time=end_time) aim = pd.DataFrame({ 'START_DATE': ['2011/01/01 00:00:00'], 'END_DATE': ['2015/01/01 00:00:00'] }) aim['START_DATE'] = pd.to_datetime(aim['START_DATE'], format='%Y/%m/%d %H:%M:%S') aim['END_DATE'] = pd.to_datetime(aim['END_DATE'], format='%Y/%m/%d %H:%M:%S') assert_frame_equal(aim, result)