diff options
| author | Scott Gasch <[email protected]> | 2022-05-29 17:53:26 -0700 |
|---|---|---|
| committer | Scott Gasch <[email protected]> | 2022-05-29 17:53:26 -0700 |
| commit | f3dbc7dc19ba5703f8f5aa9bf8af3c491b3510f6 (patch) | |
| tree | 0cc9ffe0d6c998c0ccd19983f5c514834d6459ae /dict_utils.py | |
| parent | 0fec151ee0b3596016d7a094af13085ef01a8bb4 (diff) | |
Clean up more docs to work with sphinx.
Diffstat (limited to 'dict_utils.py')
| -rw-r--r-- | dict_utils.py | 27 |
1 files changed, 20 insertions, 7 deletions
diff --git a/dict_utils.py b/dict_utils.py index 6f0f572..573e683 100644 --- a/dict_utils.py +++ b/dict_utils.py @@ -42,7 +42,6 @@ def shard(d: Dict[Any, Any], size: int) -> Iterator[Dict[Any, Any]]: """ Shards a dict into N subdicts which, together, contain all keys/values from the original unsharded dict. - """ items = d.items() for x in range(0, len(d), size): @@ -50,24 +49,35 @@ def shard(d: Dict[Any, Any], size: int) -> Iterator[Dict[Any, Any]]: def coalesce_by_creating_list(_, new_value, old_value): + """Helper for use with :meth:`coalesce` that creates a list on + collision.""" from list_utils import flatten return flatten([new_value, old_value]) def coalesce_by_creating_set(key, new_value, old_value): + """Helper for use with :meth:`coalesce` that creates a set on + collision.""" return set(coalesce_by_creating_list(key, new_value, old_value)) def coalesce_last_write_wins(_, new_value, discarded_old_value): + """Helper for use with :meth:`coalsce` that klobbers the old + with the new one on collision.""" return new_value def coalesce_first_write_wins(_, discarded_new_value, old_value): + """Helper for use with :meth:`coalsce` that preserves the old + value and discards the new one on collision.""" return old_value def raise_on_duplicated_keys(key, new_value, old_value): + """Helper for use with :meth:`coalesce` that raises an exception + when a collision is detected. + """ raise Exception(f'Key {key} is duplicated in more than one input dict.') @@ -79,10 +89,13 @@ def coalesce( """Merge N dicts into one dict containing the union of all keys / values in the input dicts. When keys collide, apply the aggregation_function which, by default, creates a list of values. - See also several other alternative functions for coalescing values - (coalesce_by_creating_set, coalesce_first_write_wins, - coalesce_last_write_wins, raise_on_duplicated_keys) or provide a - custom helper function. + See also several other alternative functions for coalescing values: + + * :meth:`coalesce_by_creating_set` + * :meth:`coalesce_first_write_wins` + * :meth:`coalesce_last_write_wins` + * :meth:`raise_on_duplicated_keys` + * or provive your own collision resolution code. >>> a = {'a': 1, 'b': 2} >>> b = {'b': 1, 'c': 2, 'd': 3} @@ -111,7 +124,7 @@ def coalesce( def item_with_max_value(d: Dict[Any, Any]) -> Tuple[Any, Any]: - """Returns the key and value with the max value in a dict. + """Returns the key and value of the item with the max value in a dict. >>> d = {'a': 1, 'b': 2, 'c': 3} >>> item_with_max_value(d) @@ -126,7 +139,7 @@ def item_with_max_value(d: Dict[Any, Any]) -> Tuple[Any, Any]: def item_with_min_value(d: Dict[Any, Any]) -> Tuple[Any, Any]: - """Returns the key and value with the min value in a dict. + """Returns the key and value of the item with the min value in a dict. >>> d = {'a': 1, 'b': 2, 'c': 3} >>> item_with_min_value(d) |
