python 计算两个角度之间的差值（结果在-pi到pi之间）

import math
def diff(a, b):
  error = a - b
  if error < -math.pi:
    return error + 2*math.pi#小于-pi加2pi
  elif error >= math.pi:
    return error - 2*math.pi#大于pi减2pi
  else:
    return error

print diff(5, -200)
print diff(math.pi, -math.pi + 0.001)