add test for generative sampling