Fix tests for the feature of label-specific flags

anmsajedulalam · May 15, 2019 · 05ed24c · 05ed24c
1 parent 7ad8d5c
commit 05ed24c
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 7 deletions.
diff --git a/tests/test_app.py b/tests/test_app.py
@@ -56,7 +56,7 @@ def check_imageData():
  (986, 742),
  (1184, 102),
  ]
- shape = label, points, None, None, 'polygon'
+ shape = label, points, None, None, 'polygon', {}
  shapes = [shape]
  win.loadLabels(shapes)
  win.saveFile()

diff --git a/tests/widgets_tests/test_label_dialog.py b/tests/widgets_tests/test_label_dialog.py
@@ -59,8 +59,9 @@ def interact():
  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 interact():
  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 interact():
  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 == {}