Explorar el Código

Fix tests for the feature of label-specific flags

Kentaro Wada hace 6 años
padre
commit
05ed24c114
Se han modificado 2 ficheros con 10 adiciones y 7 borrados
  1. 1 1
      tests/test_app.py
  2. 9 6
      tests/widgets_tests/test_label_dialog.py

+ 1 - 1
tests/test_app.py

@@ -56,7 +56,7 @@ def test_MainWindow_annotate_jpg(qtbot):
         (986, 742),
         (1184, 102),
     ]
-    shape = label, points, None, None, 'polygon'
+    shape = label, points, None, None, 'polygon', {}
     shapes = [shape]
     win.loadLabels(shapes)
     win.saveFile()

+ 9 - 6
tests/widgets_tests/test_label_dialog.py

@@ -59,8 +59,9 @@ def test_LabelDialog_popUp(qtbot):
         qtbot.keyClick(widget.edit, QtCore.Qt.Key_Enter)  # NOQA
 
     QtCore.QTimer.singleShot(500, interact)
-    text = widget.popUp('cat')
-    assert text == 'person'
+    label, flags = widget.popUp('cat')
+    assert label == 'person'
+    assert flags == {}
 
     # popUp()
 
@@ -69,8 +70,9 @@ def test_LabelDialog_popUp(qtbot):
         qtbot.keyClick(widget.edit, QtCore.Qt.Key_Enter)  # NOQA
 
     QtCore.QTimer.singleShot(500, interact)
-    text = widget.popUp()
-    assert text == 'person'
+    label, flags = widget.popUp()
+    assert label == 'person'
+    assert flags == {}
 
     # popUp() + key_Up
 
@@ -80,5 +82,6 @@ def test_LabelDialog_popUp(qtbot):
         qtbot.keyClick(widget.edit, QtCore.Qt.Key_Enter)  # NOQA
 
     QtCore.QTimer.singleShot(500, interact)
-    text = widget.popUp()
-    assert text == 'dog'
+    label, flags = widget.popUp()
+    assert label == 'dog'
+    assert flags == {}