Fix a minor bug on python test code template

This commit fixes a minor typo on python test code, which was introduced in ca1d2c491.