generator_solution.py and test_generator.py