def sort_by_time(resource): from ocgis.util.helpers import get_sorted_uris_by_time_dimension if type(resource) == list and len(resource) > 1: sorted_list = get_sorted_uris_by_time_dimension(resource) elif type(resource) == str: sorted_list = [resource] else: sorted_list = resource return sorted_list
def sort_by_time(resource): from ocgis.util.helpers import get_sorted_uris_by_time_dimension if type(resource) is list and len(resource) > 1: sorted_list = get_sorted_uris_by_time_dimension(resource) elif type(resource) is str: sorted_list = [resource] else: sorted_list = resource return sorted_list
def test_get_sorted_uris_by_time_dimension(self): rd_2001 = self.test_data.get_rd('cancm4_tasmax_2001') rd_2011 = self.test_data.get_rd('cancm4_tasmax_2011') not_sorted = [rd_2011.uri, rd_2001.uri] actual = ['tasmax_day_CanCM4_decadal2000_r2i1p1_20010101-20101231.nc', 'tasmax_day_CanCM4_decadal2010_r2i1p1_20110101-20201231.nc'] for variable in [None, 'tasmax']: is_sorted = get_sorted_uris_by_time_dimension(not_sorted, variable=variable) to_test = [os.path.split(ii)[1] for ii in is_sorted] self.assertEqual(actual, to_test)
def sort_by_time(resource): """ Sort a list of files by their time variable. :param resource: File path. :return: Sorted file list. """ from ocgis.util.helpers import get_sorted_uris_by_time_dimension if type(resource) == list and len(resource) > 1: sorted_list = get_sorted_uris_by_time_dimension(resource) elif type(resource) == str: sorted_list = [resource] else: sorted_list = resource return sorted_list