Add tests for sample and pick generators