Test Python code