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