Python CSVImportDialog示例

示例#1

文件： MainController.py 项目： jopohl/urh

    def __import_csv(self, file_name, group_id=0):
        def on_data_imported(complex_file, sample_rate):
            sample_rate = None if sample_rate == 0 else sample_rate
            self.add_files([complex_file], group_id=group_id, enforce_sample_rate=sample_rate)

        dialog = CSVImportDialog(file_name, parent=self)
        dialog.data_imported.connect(on_data_imported)
        dialog.exec_()

示例#2

文件： MainController.py 项目： MrBurne/urh

    def __import_csv(self, file_name, group_id=0):
        def on_data_imported(complex_file, sample_rate):
            sample_rate = None if sample_rate == 0 else sample_rate
            self.add_files([complex_file], group_id=group_id, enforce_sample_rate=sample_rate)

        dialog = CSVImportDialog(file_name, parent=self)
        dialog.data_imported.connect(on_data_imported)
        dialog.exec_()

示例#3

文件： test_csv_import_dialog.py 项目： starling021/uh

    def setUp(self):
        super().setUp()
        self.dialog = CSVImportDialog()

        if self.SHOW:
            self.dialog.show()

        self.i_column = self.dialog.COLUMNS["I"]
        self.q_column = self.dialog.COLUMNS["Q"]
        self.t_column = self.dialog.COLUMNS["T"]

示例#4

文件： test_csv_import_dialog.py 项目： jopohl/urh

    def setUp(self):
        super().setUp()
        self.dialog = CSVImportDialog()

        if self.SHOW:
            self.dialog.show()

        self.i_column = self.dialog.COLUMNS["I"]
        self.q_column = self.dialog.COLUMNS["Q"]
        self.t_column = self.dialog.COLUMNS["T"]

示例#5

文件： test_csv_import_dialog.py 项目： jopohl/urh

class TestCSVImportDialog(QtTestCase):
    def setUp(self):
        super().setUp()
        self.dialog = CSVImportDialog()

        if self.SHOW:
            self.dialog.show()

        self.i_column = self.dialog.COLUMNS["I"]
        self.q_column = self.dialog.COLUMNS["Q"]
        self.t_column = self.dialog.COLUMNS["T"]

    def test_invalid_file(self):
        if self.SHOW:
            self.assertTrue(self.dialog.ui.labelFileNotFound.isVisible())

        self.dialog.ui.lineEditFilename.setText("/this/file/does/not/exist")
        self.dialog.ui.lineEditFilename.editingFinished.emit()
        self.assertEqual(self.dialog.ui.plainTextEditFilePreview.toPlainText(), "")
        self.assertEqual(self.dialog.ui.tableWidgetPreview.rowCount(), 0)

    def test_comma_separated_file(self):
        filename = os.path.join(tempfile.gettempdir(), "comma.csv")
        with open(filename, "w") as f:
            f.write("this is a comment\n")
            f.write("format is\n")
            f.write("Timestamp I Q Trash\n")

            for i in range(150):
                f.write("{},{},{},{}\n".format(i / 1e6, i, random.uniform(0, 1), 42 * i))

        self.dialog.ui.lineEditFilename.setText(filename)
        self.dialog.ui.lineEditFilename.editingFinished.emit()

        self.dialog.ui.spinBoxIDataColumn.setValue(2)
        self.dialog.ui.spinBoxTimestampColumn.setValue(1)
        self.dialog.ui.spinBoxQDataColumn.setValue(3)

        for i in range(3):
            for j in range(3):
                self.assertEqual(self.dialog.ui.tableWidgetPreview.item(i, j).text(), "Invalid")

        file_preview = self.dialog.ui.plainTextEditFilePreview.toPlainText()
        self.assertEqual(len(file_preview.split("\n")), 100)

        self.assertEqual(self.dialog.ui.tableWidgetPreview.rowCount(), 100)
        self.assertEqual(self.dialog.ui.tableWidgetPreview.item(3, self.i_column).text(), "0.0")
        self.assertEqual(self.dialog.ui.tableWidgetPreview.item(99, self.i_column).text(), "96.0")

        last_preview_line = file_preview.split("\n")[-1]
        t, i, q, _ = map(float, last_preview_line.split(","))
        self.assertEqual(self.dialog.ui.tableWidgetPreview.item(99, self.i_column).text(), str(i))
        self.assertEqual(self.dialog.ui.tableWidgetPreview.item(99, self.q_column).text(), str(q))
        self.assertEqual(self.dialog.ui.tableWidgetPreview.item(99, self.t_column).text(), str(t))

    def test_semicolon_separated_file(self):
        filename = os.path.join(tempfile.gettempdir(), "semicolon.csv")
        with open(filename, "w") as f:
            f.write("I;Trash\n")

            for i in range(20):
                f.write("{};{}\n".format(i, 24 * i))

        self.dialog.ui.lineEditFilename.setText(filename)
        self.dialog.ui.lineEditFilename.editingFinished.emit()
        self.dialog.ui.comboBoxCSVSeparator.setCurrentText(";")

        self.assertTrue(self.dialog.ui.tableWidgetPreview.isColumnHidden(self.t_column))
        self.assertEqual(self.dialog.ui.tableWidgetPreview.rowCount(), 21)
        self.assertEqual(self.dialog.ui.tableWidgetPreview.item(0, self.i_column).text(), "Invalid")
        self.assertEqual(self.dialog.ui.tableWidgetPreview.item(0, self.q_column).text(), "Invalid")
        self.assertEqual(self.dialog.ui.tableWidgetPreview.item(1, self.i_column).text(), "0.0")
        self.assertEqual(self.dialog.ui.tableWidgetPreview.item(1, self.q_column).text(), "0.0")
        self.assertEqual(self.dialog.ui.tableWidgetPreview.item(2, self.i_column).text(), "1.0")
        self.assertEqual(self.dialog.ui.tableWidgetPreview.item(2, self.q_column).text(), "0.0")

        file_preview = self.dialog.ui.plainTextEditFilePreview.toPlainText()
        last_preview_line = file_preview.split("\n")[-1]
        i, _ = map(float, last_preview_line.split(";"))
        self.assertEqual(self.dialog.ui.tableWidgetPreview.item(20, self.i_column).text(), str(i))
        self.assertEqual(self.dialog.ui.tableWidgetPreview.item(20, self.q_column).text(), "0.0")

示例#6

文件： test_csv_import_dialog.py 项目： starling021/uh

class TestCSVImportDialog(QtTestCase):
    def setUp(self):
        super().setUp()
        self.dialog = CSVImportDialog()

        if self.SHOW:
            self.dialog.show()

        self.i_column = self.dialog.COLUMNS["I"]
        self.q_column = self.dialog.COLUMNS["Q"]
        self.t_column = self.dialog.COLUMNS["T"]

    def test_invalid_file(self):
        if self.SHOW:
            self.assertTrue(self.dialog.ui.labelFileNotFound.isVisible())

        self.dialog.ui.lineEditFilename.setText("/this/file/does/not/exist")
        self.dialog.ui.lineEditFilename.editingFinished.emit()
        self.assertEqual(self.dialog.ui.plainTextEditFilePreview.toPlainText(),
                         "")
        self.assertEqual(self.dialog.ui.tableWidgetPreview.rowCount(), 0)

    def test_comma_separated_file(self):
        filename = os.path.join(tempfile.gettempdir(), "comma.csv")
        with open(filename, "w") as f:
            f.write("this is a comment\n")
            f.write("format is\n")
            f.write("Timestamp I Q Trash\n")

            for i in range(150):
                f.write("{},{},{},{}\n".format(i / 1e6, i,
                                               random.uniform(0, 1), 42 * i))

        self.dialog.ui.lineEditFilename.setText(filename)
        self.dialog.ui.lineEditFilename.editingFinished.emit()

        self.dialog.ui.spinBoxIDataColumn.setValue(2)
        self.dialog.ui.spinBoxTimestampColumn.setValue(1)
        self.dialog.ui.spinBoxQDataColumn.setValue(3)

        for i in range(3):
            for j in range(3):
                self.assertEqual(
                    self.dialog.ui.tableWidgetPreview.item(i, j).text(),
                    "Invalid")

        file_preview = self.dialog.ui.plainTextEditFilePreview.toPlainText()
        self.assertEqual(len(file_preview.split("\n")), 100)

        self.assertEqual(self.dialog.ui.tableWidgetPreview.rowCount(), 100)
        self.assertEqual(
            self.dialog.ui.tableWidgetPreview.item(3, self.i_column).text(),
            "0.0")
        self.assertEqual(
            self.dialog.ui.tableWidgetPreview.item(99, self.i_column).text(),
            "96.0")

        last_preview_line = file_preview.split("\n")[-1]
        t, i, q, _ = map(float, last_preview_line.split(","))
        self.assertEqual(
            self.dialog.ui.tableWidgetPreview.item(99, self.i_column).text(),
            str(i))
        self.assertEqual(
            self.dialog.ui.tableWidgetPreview.item(99, self.q_column).text(),
            str(q))
        self.assertEqual(
            self.dialog.ui.tableWidgetPreview.item(99, self.t_column).text(),
            str(t))

    def test_semicolon_separated_file(self):
        filename = os.path.join(tempfile.gettempdir(), "semicolon.csv")
        with open(filename, "w") as f:
            f.write("I;Trash\n")

            for i in range(20):
                f.write("{};{}\n".format(i, 24 * i))

        self.dialog.ui.lineEditFilename.setText(filename)
        self.dialog.ui.lineEditFilename.editingFinished.emit()
        self.dialog.ui.comboBoxCSVSeparator.setCurrentText(";")

        self.assertTrue(
            self.dialog.ui.tableWidgetPreview.isColumnHidden(self.t_column))
        self.assertEqual(self.dialog.ui.tableWidgetPreview.rowCount(), 21)
        self.assertEqual(
            self.dialog.ui.tableWidgetPreview.item(0, self.i_column).text(),
            "Invalid")
        self.assertEqual(
            self.dialog.ui.tableWidgetPreview.item(0, self.q_column).text(),
            "Invalid")
        self.assertEqual(
            self.dialog.ui.tableWidgetPreview.item(1, self.i_column).text(),
            "0.0")
        self.assertEqual(
            self.dialog.ui.tableWidgetPreview.item(1, self.q_column).text(),
            "0.0")
        self.assertEqual(
            self.dialog.ui.tableWidgetPreview.item(2, self.i_column).text(),
            "1.0")
        self.assertEqual(
            self.dialog.ui.tableWidgetPreview.item(2, self.q_column).text(),
            "0.0")

        file_preview = self.dialog.ui.plainTextEditFilePreview.toPlainText()
        last_preview_line = file_preview.split("\n")[-1]
        i, _ = map(float, last_preview_line.split(";"))
        self.assertEqual(
            self.dialog.ui.tableWidgetPreview.item(20, self.i_column).text(),
            str(i))
        self.assertEqual(
            self.dialog.ui.tableWidgetPreview.item(20, self.q_column).text(),
            "0.0")