Add a generator test