add test for choice generator