Python program testing