code: def _is_trivial_angle(rad: float, atol: float) -> bool:
      
      return abs(rad) < atol or abs(abs(rad) - np.pi / 4) < atol
docs: Tests if a circuit for an operator exp(i*rad*XX) (or YY, or ZZ) can
      be performed with a whole CZ.
  
      Args:
          rad: The angle in radians, assumed to be in the range [-pi/4, pi/4]