Test code in python