Fixed tests for draw_samples()