Python DataFrameModel.columnCount Examples

Example #1

File: test_dataframeeditor.py Project: G-VAR/spyder

def test_dataframemodel_basic():
    df = DataFrame({'colA': [1, 3], 'colB': ['c', 'a']})
    dfm = DataFrameModel(df)
    assert dfm.rowCount() == 2
    assert dfm.columnCount() == 3
    assert dfm.data(dfm.createIndex(0, 0)) == '0'
    assert dfm.data(dfm.createIndex(0, 1)) == '1'
    assert dfm.data(dfm.createIndex(0, 2)) == 'c'
    assert dfm.data(dfm.createIndex(1, 0)) == '1'
    assert dfm.data(dfm.createIndex(1, 1)) == '3'
    assert dfm.data(dfm.createIndex(1, 2)) == 'a'

Example #2

def test_dataframemodel_basic():
    df = DataFrame({'colA': [1, 3], 'colB': ['c', 'a']})
    dfm = DataFrameModel(df)
    assert dfm.rowCount() == 2
    assert dfm.columnCount() == 3
    assert dfm.data(dfm.createIndex(0, 0)) == '0'
    assert dfm.data(dfm.createIndex(0, 1)) == '1'
    assert dfm.data(dfm.createIndex(0, 2)) == 'c'
    assert dfm.data(dfm.createIndex(1, 0)) == '1'
    assert dfm.data(dfm.createIndex(1, 1)) == '3'
    assert dfm.data(dfm.createIndex(1, 2)) == 'a'