def test_flatten_record_ranges(self, uow): syrx_file_handler = SyrxFileHandler() records = [RecordRange("syrx_num_1", datetime.datetime(2012, 1, 1), datetime.datetime(2012, 3, 1)), RecordRange("syrx_num_1", datetime.datetime(2012, 4, 1), datetime.datetime(2012, 6, 1)), RecordRange("syrx_num_2", datetime.datetime(2012, 7, 1), datetime.datetime(2012, 9, 1)), RecordRange("syrx_num_2", datetime.datetime(2012, 10, 1), datetime.datetime(2012, 12, 1))] rv = syrx_file_handler.flatten_record_ranges(records) assert rv[0].syrx_num == "syrx_num_1" assert rv[0].start_date == datetime.datetime(2012, 1, 1) assert rv[0].end_date == datetime.datetime(2012, 6, 1) assert rv[1].syrx_num == "syrx_num_2" assert rv[1].start_date == datetime.datetime(2012, 7, 1) assert rv[1].end_date == datetime.datetime(2012, 12, 1)