python basic test code