update in multimap function in utils.py and added test for it (#1014)

* update in multimap functi n in utils.py and added test for it

* made changes according to PEP style

* broke the long sentence to 2 shorter sentences