add sampling tests