np.testing

 np.testing.assert_allclose(a, b, atol=1e-5

  atol=1e-5  # absolute error: |a - b| < 1e-5
  rtol=1e-3  # relative error: |a - b| < 1e-3 * |b|