Naming Conventions

Suggestion: Renaming function main to run_tests

It may be a python convention to name functions main when they should be called if the module/script is executed directly ...

if __name__ == "__main__":
    main()

... but main says nothing about what happens.

if __name__ == "__main__":
    run_tests()

This gives the reader a clear idea about what happens.