Create corpus - fix tests

biolab · Mar 7, 2023 · ed68d54 · ed68d54
1 parent 9e795be
commit ed68d54
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/orangecontrib/text/widgets/tests/test_owcreatecorpus.py b/orangecontrib/text/widgets/tests/test_owcreatecorpus.py
@@ -28,29 +28,29 @@ def test_add_remove_editors(self):
         self.assertEqual(5, len(self.widget.texts))
         self.assertListEqual([("", "")] * 5, self.widget.texts)
 
-        # click any button in editor_vbox - there are only buttons for removing
-        self.widget.editor_vbox.findChild(QPushButton).click()
+        # click x button for first editor
+        self.widget.editors[0].findChild(QPushButton).click()
         self.assertEqual(4, len(self.widget.editors))
         self.assertEqual(4, len(self.widget.texts))
         self.assertListEqual([("", "")] * 4, self.widget.texts)
 
-        self.widget.editor_vbox.findChild(QPushButton).click()
+        self.widget.editors[0].findChild(QPushButton).click()
         self.assertEqual(3, len(self.widget.editors))
         self.assertEqual(3, len(self.widget.texts))
         self.assertListEqual([("", "")] * 3, self.widget.texts)
 
-        self.widget.editor_vbox.findChild(QPushButton).click()
+        self.widget.editors[0].findChild(QPushButton).click()
         self.assertEqual(2, len(self.widget.editors))
         self.assertEqual(2, len(self.widget.texts))
         self.assertListEqual([("", "")] * 2, self.widget.texts)
 
-        self.widget.editor_vbox.findChild(QPushButton).click()
+        self.widget.editors[0].findChild(QPushButton).click()
         self.assertEqual(1, len(self.widget.editors))
         self.assertEqual(1, len(self.widget.texts))
         self.assertListEqual([("", "")], self.widget.texts)
 
         # last editor cannot be removed
-        self.widget.editor_vbox.findChild(QPushButton).click()
+        self.widget.editors[0].findChild(QPushButton).click()
         self.assertEqual(1, len(self.widget.editors))
         self.assertEqual(1, len(self.widget.texts))
         self.assertListEqual([("", "")], self.widget.texts)