Rename test_generator to test_generator.py