From ddc1f803c0ae256e80a63998b6db6f4fc1991d4c Mon Sep 17 00:00:00 2001 From: Florian Bruhin Date: Sun, 8 May 2016 20:00:34 +0200 Subject: [PATCH] Clean up assertion --- tests/unit/completion/test_sortfilter.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) 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)