ENH: added timer during test to click resampling "yes" button automat…

…ically
QIICR · Oct 16, 2017 · 044c8fa · 044c8fa
1 parent 3ca032d
commit 044c8fa
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/Testing/QuantitativeReportingTests.py b/Testing/QuantitativeReportingTests.py
@@ -172,15 +172,28 @@ def test_import_labelmap(self):
         labels.append(label)
 
     labelImportWidget = qrWidget.labelMapImportWidget
+
+    timer = qt.QTimer()
+    timer.setInterval(2000)
+    timer.timeout.connect(self._checkFocusAndClickButton)
+    timer.start()
+
     for label in labels:
       labelImportWidget.labelMapSelector.setCurrentNode(label)
       labelImportWidget.importButton.click()
 
+    timer.stop()
+
     segmentation = qrWidget.segmentEditorWidget.segmentationNode.GetSegmentation()
     self.assertEquals(segmentation.GetNumberOfSegments(), len(labels))
 
     self.delayDisplay('Test passed!')
 
+  def _checkFocusAndClickButton(self):
+    focus = slicer.app.focusWidget()
+    if type(focus) is qt.QPushButton:
+      focus.click()
+
   def test_import_segmentation(self):
     self.delayDisplay('Starting %s' % inspect.stack()[0][3])