def test_customDTypeModel_check_init(self):
model = ColumnDtypeModel()
assert model.dataFrame().empty == True
assert model.editable() == False
model = ColumnDtypeModel(editable=True)
assert model.editable() == True
def setDataFrame(self, dataFrame, copyDataFrame=False):
"""setter function to _dataFrame. Holds all data.
Note:
It's not implemented with python properties to keep Qt conventions.
Raises:
TypeError: if dataFrame is not of type pandas.core.frame.DataFrame.
Args:
dataFrame (pandas.core.frame.DataFrame): assign dataFrame to _dataFrame. Holds all the data displayed.
copyDataFrame (bool, optional): create a copy of dataFrame or use it as is. defaults to False.
If you use it as is, you can change it from outside otherwise you have to reset the dataFrame
after external changes.
"""
if not isinstance(dataFrame, pandas.core.frame.DataFrame):
raise TypeError("not of type pandas.core.frame.DataFrame")
self.layoutAboutToBeChanged.emit()
if copyDataFrame:
self._dataFrame = dataFrame.copy()
else:
self._dataFrame = dataFrame
self._columnDtypeModel = ColumnDtypeModel(dataFrame)
self._columnDtypeModel.dtypeChanged.connect(self.propagateDtypeChanges)
# self._columnDtypeModel.changingDtypeFailed.connect(
# lambda columnName, index, dtype: self.changingDtypeFailed.emit(columnName, index, dtype)
# )
self.layoutChanged.emit()
def test_setDataFrame(self, dataframe):
model = ColumnDtypeModel()
model.setDataFrame(dataframe)
assert model.rowCount(5)
with pytest.raises(TypeError) as err:
model.setDataFrame(['some', 'neat', 'list', 'entries'])
assert 'not of type pandas.core.frame.DataFrame' in str(err.value)
def test_flags(self, dataframe):
model = ColumnDtypeModel(dataFrame=dataframe)
model.setEditable(True)
index = model.index(0, 0)
assert model.flags(index) == Qt.ItemIsEnabled | Qt.ItemIsSelectable
index = index.sibling(0, 1)
assert model.flags(index) == Qt.ItemIsSelectable | Qt.ItemIsEnabled | Qt.ItemIsEditable
index = index.sibling(15, 1)
assert model.flags(index) == Qt.NoItemFlags
def test_headerData(self):
model = ColumnDtypeModel()
ret = model.headerData(0, Qt.Horizontal)
assert ret == 'column'
ret = model.headerData(1, Qt.Horizontal)
assert ret == 'data type'
ret = model.headerData(2, Qt.Horizontal)
assert ret == None
ret = model.headerData(0, Qt.Horizontal, Qt.EditRole)
assert ret == None
ret = model.headerData(0, Qt.Vertical)
assert ret == None
def test_setData(self, dataframe, language_values, qtbot):
model = ColumnDtypeModel(dataFrame=dataframe)
index = model.index(3, 1)
model.setEditable(True)
# change all values except datetime
datetime = ()
for (expected_type, string) in language_values:
if expected_type == numpy.dtype('<M8[ns]'):
datetime = (string, expected_type)
continue
else:
model.setData(index, string)
assert index.data(DTYPE_ROLE) == expected_type
assert model.setData(index, 'bool', Qt.DisplayRole) == False
with pytest.raises(Exception) as err:
model.setData(index, datetime[0])
assert "Can't convert a boolean value into a datetime value" in str(err.value)
# rewrite this with parameters
for data in [
["2012-12-13"],
["2012-12-13 19:10"],
["2012-12-13 19:10:10"]
]:
df = pandas.DataFrame(data, columns=["datetime"])
model = ColumnDtypeModel(dataFrame=df)
index = model.index(0, 0)
model.setEditable(True)
assert model.setData(index, "date and time") == True
# convert datetime to anything else does not work and leave the
# datatype unchanged. An error message is emitted.
with qtbot.waitSignal(model.changeFailed):
model.setData(index, 'bool')
def test_editing(self, dataframe, qtbot):
model = ColumnDtypeModel(dataFrame=dataframe)
model.setEditable(True)
tableView = QtGui.QTableView()
qtbot.addWidget(tableView)
tableView.setModel(model)
delegate = DtypeComboDelegate(tableView)
tableView.setItemDelegateForColumn(1, delegate)
tableView.show()
index = model.index(0, 1)
preedit_data = index.data(DTYPE_ROLE)
tableView.edit(index)
editor = tableView.findChildren(QtGui.QComboBox)[0]
selectedIndex = editor.currentIndex()
editor.setCurrentIndex(selectedIndex+1)
postedit_data = index.data(DTYPE_ROLE)
assert preedit_data != postedit_data
def test_data(self, dataframe):
model = ColumnDtypeModel(dataFrame=dataframe)
index = model.index(0, 0)
# get data for display role
ret = index.data()
assert ret == 'Foo'
# edit role does the same as display role
ret = index.data(Qt.EditRole)
assert ret == 'Foo'
# datatype only defined for column 1
ret = index.data(DTYPE_ROLE)
assert ret == None
# datatype column
index = index.sibling(0, 1)
ret = index.data(DTYPE_ROLE)
assert ret == numpy.dtype(numpy.int64)
# check translation / display text
assert index.data(
) == 'integer (64 bit)' == SupportedDtypes.description(ret)
# column not defined
index = index.sibling(0, 2)
assert index.data(DTYPE_ROLE) == None
# invalid index
index = QtCore.QModelIndex()
assert model.data(index) == None
index = model.index(2, 0)
# get data for display role
ret = index.data()
assert ret == 'Spam'
def test_setData(self, dataframe, language_values, qtbot):
model = ColumnDtypeModel(dataFrame=dataframe)
index = model.index(3, 1)
model.setEditable(True)
# change all values except datetime
datetime = ()
for (expected_type, string) in language_values:
if expected_type == numpy.dtype('<M8[ns]'):
datetime = (string, expected_type)
continue
else:
model.setData(index, string)
assert index.data(DTYPE_ROLE) == expected_type
assert model.setData(index, 'bool', Qt.DisplayRole) == False
# change datatype to datetime
assert model.setData(index, datetime[0]) == True
# convert datetime to anything else does not work and leave the
# datatype unchanged. An error message is emitted.
with qtbot.waitSignal(model.changeFailed):
model.setData(index, 'bool')
def test_rowCount(self, dataframe):
model = ColumnDtypeModel()
assert model.rowCount() == 0
model.setDataFrame(dataframe)
assert model.rowCount(5)
def test_columnCount(self):
model = ColumnDtypeModel()
assert model.columnCount() == 2
