Clean up assertion

2016-05-08 20:00:34 +02:00 · 2016-05-08 20:00:34 +02:00 · ddc1f803c0
commit ddc1f803c0
parent f49cc4e901
1 changed files with 2 additions and 4 deletions
--- 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)