diff --git a/tests/unit/completion/test_sortfilter.py b/tests/unit/completion/test_sortfilter.py index c743db87e..86a132e1d 100644 --- a/tests/unit/completion/test_sortfilter.py +++ b/tests/unit/completion/test_sortfilter.py @@ -44,7 +44,5 @@ def test_filter_accepts_row(pattern, data, expected): idx = filter_model.index(0, 0) assert idx.isValid() - if expected: - assert filter_model.rowCount(idx) == 1 - else: - assert filter_model.rowCount(idx) == 0 + row_count = filter_model.rowCount(idx) + assert row_count == (1 if expected else 0)