[Fix] Fix warning capture

fanqiNO1 · Feb 18, 2024 · c51a235 · c51a235
1 parent 5b64bee
commit c51a235
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/tests/test_utils/test_dl_utils/test_torch_ops.py b/tests/test_utils/test_dl_utils/test_torch_ops.py
@@ -1,14 +1,15 @@
 # Copyright (c) OpenMMLab. All rights reserved.
-import pytest
+import warnings
+
 import torch
 
 from mmengine.utils.dl_utils import torch_meshgrid
 
 
 def test_torch_meshgrid():
     # torch_meshgrid should not throw warning
-    with pytest.warns(UserWarning) as record:
+    with warnings.catch_warnings():
+        warnings.simplefilter('error')
         x = torch.tensor([1, 2, 3])
         y = torch.tensor([4, 5, 6])
         grid_x, grid_y = torch_meshgrid(x, y)
-    assert len(record) == 0