Add main in test

[zhhsplendid]

I forgot to add main in the python test.

[paddle-bot-old[bot]]

Thanks for your contribution!