"...git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "34c90dbb31f4956e72086ac43ca7d8154c2aadae"
Unverified Commit 57e470ab authored by xiliu8006's avatar xiliu8006 Committed by GitHub
Browse files

[Enhance] Confirm the input does not be changed in bg_point_filter unit test (#621)

* add unittest to confirm the input is not be modified

* assert gt_bboxes_3d

* rm .swag

* modify np.allclose to np.equal

* fix error tap

* fix np.equal() error
parent ed090ed2
......@@ -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)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment