Test CompletionFilterModel.count
This commit is contained in:
parent
f19b818b58
commit
8e7002db7a
@ -93,3 +93,24 @@ def test_set_source_model():
|
||||
assert filter_model.srcmodel is model2
|
||||
assert filter_model.sourceModel() is model2
|
||||
assert not filter_model.pattern
|
||||
|
||||
@pytest.mark.parametrize('tree, expected', [
|
||||
([['Aa']], 1),
|
||||
([['Aa'], ['Ba']], 2),
|
||||
([['Aa', 'Ab', 'Ac'], ['Ba', 'Bb'], ['Ca']], 6),
|
||||
([[], ['Ba']], 1),
|
||||
([[], [], ['Ca']], 1),
|
||||
([[], [], ['Ca', 'Cb']], 2),
|
||||
([['Aa'], []], 1),
|
||||
([['Aa'], []], 1),
|
||||
([['Aa'], [], []], 1),
|
||||
([['Aa'], [], ['Ca']], 2),
|
||||
])
|
||||
def test_count(tree, expected):
|
||||
model = base.BaseCompletionModel()
|
||||
for catdata in tree:
|
||||
cat = model.new_category('')
|
||||
for name in catdata:
|
||||
model.new_item(cat, name, '')
|
||||
filter_model = sortfilter.CompletionFilterModel(model)
|
||||
assert filter_model.count() == expected
|
||||
|
Loading…
Reference in New Issue
Block a user