X-Git-Url: https://wannabe.guru.org/gitweb/?a=blobdiff_plain;f=dict_utils.py;h=b1464c6bb9967ce4efa48318a288babd0ff322e9;hb=865825894beeedd47d26dd092d40bfee582f5475;hp=7b0edb50a87a5854f20326408cfc0a03d22408dc;hpb=d08bad64a6884f25d28a2c38c6cd1c87b4335188;p=python_utils.git diff --git a/dict_utils.py b/dict_utils.py index 7b0edb5..b1464c6 100644 --- a/dict_utils.py +++ b/dict_utils.py @@ -9,7 +9,7 @@ def init_or_inc( key: Any, *, init_value: Any = 1, - inc_function: Callable[..., Any] = lambda x: x + 1 + inc_function: Callable[..., Any] = lambda x: x + 1, ) -> bool: """ Initialize a dict value (if it doesn't exist) or increments it (using the @@ -38,6 +38,7 @@ 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): @@ -46,6 +47,7 @@ def shard(d: Dict[Any, Any], size: int) -> Iterator[Dict[Any, Any]]: def coalesce_by_creating_list(key, new_value, old_value): from list_utils import flatten + return flatten([new_value, old_value]) @@ -66,9 +68,9 @@ def raise_on_duplicated_keys(key, new_value, old_value): def coalesce( - inputs: Iterator[Dict[Any, Any]], - *, - aggregation_function: Callable[[Any, Any], Any] = coalesce_by_creating_list + inputs: Iterator[Dict[Any, Any]], + *, + aggregation_function: Callable[[Any, Any], Any] = coalesce_by_creating_list, ) -> Dict[Any, Any]: """Merge N dicts into one dict containing the union of all keys / values in the input dicts. When keys collide, apply the @@ -230,4 +232,5 @@ def dict_to_key_value_lists(d: Dict[Any, Any]) -> Tuple[List[Any], List[Any]]: if __name__ == '__main__': import doctest + doctest.testmod()