Improved Python tests

- Make all test functions highlighted.
- Auto detect  Python executable from shebang