python编写函数判断三角形