def test_final_data_transform_factory_with_iterable_returns_tuple_with_no_options( x): assert _final_data_transform_factory(0, '')(iter([x]), '') == (x, ) # UNGROUPLETTERS without LOCALE does nothing, as does LOCALE without UNGROUPLETTERS assert _final_data_transform_factory(ns.UNGROUPLETTERS, '')(iter( [x]), '') == _final_data_transform_factory(0, '')(iter([x]), '') assert _final_data_transform_factory(ns.LOCALE, '')(iter( [x]), '') == _final_data_transform_factory(0, '')(iter([x]), '')
def test_final_data_transform_factory_returns_first_element_in_first_tuple_element( x, y): assume(x) assume(not isnan(y)) assume(not isinf(y)) assert _final_data_transform_factory(ns.LOCALE | ns.UNGROUPLETTERS, '')( (x, y), ''.join(map(py23_str, [x, y]))) == ((x[0], ), (x, y))
def test_final_data_transform_factory_returns_first_element_in_first_tuple_element_caseswapped_with_DUMB_and_LOWERCASEFIRST( x, y): assume(x) assume(not isnan(y)) assume(not isinf(y)) assert _final_data_transform_factory( ns.LOCALE | ns.UNGROUPLETTERS | ns._DUMB | ns.LOWERCASEFIRST, '')((x, y), ''.join(map(py23_str, [x, y]))) == ((x[0].swapcase(), ), (x, y))
def test_final_data_transform_factory_returns_first_element_in_first_tuple_element_caseswapped_with_DUMB_and_LOWERCASEFIRST_example(): assert _final_data_transform_factory(ns.LOCALE | ns.UNGROUPLETTERS | ns._DUMB | ns.LOWERCASEFIRST, '', '')(('this', 60), 'this60') == (('T',), ('this', 60))
def test_final_data_transform_factory_returns_first_element_in_first_tuple_element_example(): assert _final_data_transform_factory(ns.LOCALE | ns.UNGROUPLETTERS, '', '')(('this', 60), 'this60') == (('t',), ('this', 60))
def test_final_data_transform_factory_with_null_string_first_element_adds_empty_string_on_first_tuple_element(): assert _final_data_transform_factory(ns.LOCALE | ns.UNGROUPLETTERS, '', 'xx')(('', 60), '') == (('xx',), ('', 60))
def test_final_data_transform_factory_with_empty_tuple_returns_double_empty_tuple(): assert _final_data_transform_factory(ns.LOCALE | ns.UNGROUPLETTERS, '', '')((), '') == ((), ())
def test_final_data_transform_factory_with_iterable_returns_tuple_with_no_options_example(): assert _final_data_transform_factory(0, '', '')(iter([7]), '') == (7,)
given, ) from hypothesis.strategies import ( lists, text, floats, integers, binary, ) if PY_VERSION >= 3: long = int regex = _regex_chooser[ns.INT] pre = _input_string_transform_factory(ns.INT) post = _string_component_transform_factory(ns.INT) after = _final_data_transform_factory(ns.INT, '') string_func = _parse_string_factory(ns.INT, '', regex.split, pre, post, after) bytes_func = _parse_bytes_factory(ns.INT) num_func = _parse_number_factory(ns.INT, '') def test__natsort_key_with_numeric_input_and_PATH_returns_number_in_nested_tuple( ): # It gracefully handles as_path for numeric input by putting an extra tuple around it # so it will sort against the other as_path results. sfunc = _parse_path_factory(string_func) bytes_func = _parse_bytes_factory(ns.PATH) num_func = _parse_number_factory(ns.PATH, '') assert _natsort_key(10, None, sfunc, bytes_func, num_func) == (('', 10), )
def natsort_keygen(key=None, alg=0, **_kwargs): """\ Generate a key to sort strings and numbers naturally. Generate a key to sort strings and numbers naturally, not lexicographically. This key is designed for use as the `key` argument to functions such as the `sorted` builtin. The user may customize the generated function with the arguments to `natsort_keygen`, including an optional `key` function. Parameters ---------- key : callable, optional A key used to manipulate the input value before parsing for numbers. It is **not** applied recursively. It should accept a single argument and return a single value. alg : ns enum, optional This option is used to control which algorithm `natsort` uses when sorting. For details into these options, please see the :class:`ns` class documentation. The default is `ns.INT`. Returns ------- out : function A function that parses input for natural sorting that is suitable for passing as the `key` argument to functions such as `sorted`. See Also -------- natsorted Examples -------- `natsort_keygen` is a convenient way to create a custom key to sort lists in-place (for example).:: >>> a = ['num5.10', 'num-3', 'num5.3', 'num2'] >>> a.sort(key=natsort_keygen(alg=ns.REAL)) >>> a [{u}'num-3', {u}'num2', {u}'num5.10', {u}'num5.3'] """ # Transform old arguments to the ns enum. try: alg = _args_to_enum(**_kwargs) | alg except TypeError: msg = "natsort_keygen: 'alg' argument must be from the enum 'ns'" raise ValueError(msg+', got {0}'.format(py23_str(alg))) # Add the _DUMB option if the locale library is broken. if alg & ns.LOCALEALPHA and natsort.compat.locale.dumb_sort(): alg |= ns._DUMB # Set some variables that will be passed to the factory functions sep = natsort.compat.locale.null_string if alg & ns.LOCALEALPHA else '' regex = _regex_chooser[alg & ns._NUMERIC_ONLY] # Create the functions that will be used to split strings. input_transform = _input_string_transform_factory(alg) component_transform = _string_component_transform_factory(alg) final_transform = _final_data_transform_factory(alg, sep) # Create the high-level parsing functions for strings, bytes, and numbers. string_func = _parse_string_factory( alg, sep, regex.split, input_transform, component_transform, final_transform ) if alg & ns.PATH: string_func = _parse_path_factory(string_func) bytes_func = _parse_bytes_factory(alg) num_func = _parse_number_factory(alg, sep) # Return the natsort key with the parsing path pre-chosen. return partial( _natsort_key, key=key, string_func=string_func, bytes_func=bytes_func, num_func=num_func )