fixing syntax error main def  in test.py