diff --git a/tests/test_data/test_pipelines/test_augmentations/test_transforms_3d.py b/tests/test_data/test_pipelines/test_augmentations/test_transforms_3d.py index 1de4437376..11ca4c4ecc 100644 --- a/tests/test_data/test_pipelines/test_augmentations/test_transforms_3d.py +++ b/tests/test_data/test_pipelines/test_augmentations/test_transforms_3d.py @@ -614,6 +614,7 @@ def test_background_points_filter(): points = np.concatenate([points, extra_points.numpy()], 0) points = LiDARPoints(points, points_dim=4) input_dict = dict(points=points, gt_bboxes_3d=gt_bboxes_3d) + origin_gt_bboxes_3d = gt_bboxes_3d.clone() input_dict = background_points_filter(input_dict) points = input_dict['points'].tensor.numpy() @@ -622,7 +623,9 @@ def test_background_points_filter(): '[[0.5, 2.0, 0.5]])' assert repr_str == expected_repr_str assert points.shape == (800, 4) - assert np.allclose(orig_points, points) + assert np.equal(orig_points, points).all() + assert np.equal(input_dict['gt_bboxes_3d'].tensor.numpy(), + origin_gt_bboxes_3d.tensor.numpy()).all() # test single float config BackgroundPointsFilter(0.5)