generators: add two more tests, to handle return killing the generator