[Fix] textspotting ut (open-mmlab#1599)

serious-gist · Dec 6, 2022 · 5940d6b · 5940d6b
1 parent fa4fd1f
commit 5940d6b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_visualization/test_textspotting_visualizer.py b/tests/test_visualization/test_textspotting_visualizer.py
@@ -13,7 +13,7 @@
 from mmocr.visualization import TextSpottingLocalVisualizer
 
 
-class TestTextKIELocalVisualizer(unittest.TestCase):
+class TestTextSpottingLocalVisualizer(unittest.TestCase):
 
     def setUp(self):
         h, w = 12, 10
@@ -95,7 +95,7 @@ def test_add_datasample(self):
                 draw_pred=False,
                 out_file=out_file)
             self._assert_image_and_shape(out_file, (h, w * 2, c))
-            bboxes = self.data_sample.pred_instances.pop('bboxes')
+            bboxes = self.data_sample.pred_instances.get('bboxes')
             bboxes = bboxes.tolist()
             polys = [bbox2poly(bbox) for bbox in bboxes]
             self.data_sample.pred_instances.polygons = polys