`multinomial` optional parameter is now `replace`
update test code comment for clarity

Based on @Jorge-C's suggestions
update test code to use set instead of dict

Much clearer- thanks @Jorge-C!