Adding tests for generators.