diff --git a/tests/unit/geometry/test_vec.py b/tests/unit/geometry/test_vec.py index a587270ec..b9fa9ad05 100644 --- a/tests/unit/geometry/test_vec.py +++ b/tests/unit/geometry/test_vec.py @@ -12,6 +12,8 @@ some_int = 42 vec3d = Vec3D(1.0, 2.0, 3.0) +vec3d_fp1 = Vec3D(0.1, 0.6 / 3.0, 0.1 + 0.2) +vec3d_fp2 = Vec3D(0.1, 0.2, 0.3) vec3d_neg = Vec3D(-1.0, -2.0, -3.0) vec3d_lg = Vec3D(1.5, 2.5, 3.5) vec3d_mx = Vec3D(0.5, 2.5, 3.0) @@ -172,6 +174,10 @@ def test_neg(): assert vec3d == -vec3d_neg +def test_round(): + assert round(vec3d_fp1, 8) == vec3d_fp2 + + @pytest.mark.parametrize( "arg1, arg2, fname, dtype", [ @@ -346,6 +352,7 @@ def test_inference_return_int3d(x: Vec3D[int]): some_int * intvec3d, some_int // intvec3d, some_int % intvec3d, + round(intvec3d), ]: test_inference_return_int3d(x) @@ -402,5 +409,6 @@ def test_inference_return_vec3d(x: Vec3D): some_float // intvec3d, some_float / intvec3d, some_float % intvec3d, + round(vec3d), ]: test_inference_return_vec3d(y)