Fix imports

[mikkokotila]

No description provided.

[pep8speaks]

Hello @mikkokotila! Thanks for updating this PR. We checked the lines you've touched for PEP 8 issues, and found:

• In the file test_script.py:

Line 410:1: E265 block comment should start with '# '
Line 411:1: E265 block comment should start with '# '
Line 417:1: E265 block comment should start with '# '
Line 421:1: E265 block comment should start with '# '
Line 422:1: E265 block comment should start with '# '
Line 424:1: E265 block comment should start with '# '
Line 425:1: E265 block comment should start with '# '
Line 426:1: E265 block comment should start with '# '