add sampling tests with weighted-one-of