cleaned up SampleGenerator's test