test_datawindow3.py 文件源码

def test_pressing_select_diameter_button_will_select_items_with_current_diameter(self):
        self.create_a_new_project()
        from PyQt5.QtTest import QTest
        from PyQt5.QtCore import Qt
        self.aw.datawindow.ui.select_diameter_combobox.setCurrentIndex(3)
        self.assertEqual(self.aw.datawindow.ui.select_diameter_combobox.currentText(), '10.0')
        QTest.mouseClick(self.aw.datawindow.ui.select_diameter_button, Qt.LeftButton)
        nw = self.aw.networkmap.get_plot().get_selected_items()
        sel2 = [x.id_ for x in nw if (hasattr(x, "id_"))]
        dw = self.aw.datawindow.get_plot().get_selected_items()
        sel = [x.id_ for x in dw if (hasattr(x, "id_"))]
        self.assertSetEqual(set(sel), set(['12', '21', '111']))
        self.assertSetEqual(set(sel), set(sel2))
        self.aw.datawindow.ui.select_diameter_combobox.setCurrentIndex(2)
        self.assertEqual(self.aw.datawindow.ui.select_diameter_combobox.currentText(), '8.0')

        QTest.mouseClick(self.aw.datawindow.ui.select_diameter_button, Qt.LeftButton)
        nw = self.aw.networkmap.get_plot().get_selected_items()
        sel2 = [x.id_ for x in nw if (hasattr(x, "id_"))]
        dw = self.aw.datawindow.get_plot().get_selected_items()
        sel = [x.id_ for x in dw if (hasattr(x, "id_"))]
        self.assertSetEqual(set(sel), set(['113', '121']))
        time.sleep(1)