def test_one_id_gencondata_start_date_before_all(self):
start_time = datetime.strptime('2010/01/01 01:00:00',
'%Y/%m/%d %H:%M:%S')
table_name = 'dummy'
defaults.primary_date_columns['dummy'] = 'EFFECTIVEDATE'
defaults.effective_date_group_col['dummy'] = ['GENCONID']
result = query_wrapers.most_recent_records_before_start_time(self.dummyGenConData, start_time, table_name).\
sort_values('EFFECTIVEDATE', ascending=False).reset_index(drop=True)
aim = pd.DataFrame({'EFFECTIVEDATE': ['2017/01/01 00:00:00', '2017/01/04 00:15:00', '2018/05/01 00:00:00'],
'VERSIONNO': ['5', '1', '1'],
'GENCONID': ['ID1', 'ID1', 'ID1']}).sort_values('EFFECTIVEDATE', ascending=False)\
.reset_index(drop=True)
aim['EFFECTIVEDATE'] = pd.to_datetime(aim['EFFECTIVEDATE'],
format='%Y/%m/%d %H:%M:%S')
assert_frame_equal(aim, result)
def test_2_id_and_repeated_effectivedate_start_date_after_all(self):
start_time = datetime.strptime('2019/06/01 00:00:00',
'%Y/%m/%d %H:%M:%S')
table_name = 'dummy'
defaults.primary_date_columns['dummy'] = 'EFFECTIVEDATE'
defaults.effective_date_group_col['dummy'] = ['GENCONID']
result = query_wrapers.most_recent_records_before_start_time(
self.dummyGenConData3, start_time,
table_name).reset_index(drop=True)
aim = pd.DataFrame({
'EFFECTIVEDATE': [
'2017/01/04 00:15:00', '2017/01/04 00:15:00',
'2017/01/04 00:15:00', '2017/01/04 00:15:00'
],
'VERSIONNO': ['1', '2', '1', '2'],
'GENCONID': ['ID1', 'ID1', 'ID2', 'ID2']
})
aim['EFFECTIVEDATE'] = pd.to_datetime(aim['EFFECTIVEDATE'],
format='%Y/%m/%d %H:%M:%S')
assert_frame_equal(aim, result)
