From 81f5b7115f6600ad8ba178c4b2b1a74f97120b94 Mon Sep 17 00:00:00 2001 From: Ryan Roden-Corrent Date: Wed, 5 Jul 2017 08:44:56 -0400 Subject: [PATCH] Add spec=[] to two mock functions in tests. --- tests/unit/completion/test_completionmodel.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/unit/completion/test_completionmodel.py b/tests/unit/completion/test_completionmodel.py index bc7f15ab8..2c0af3a4b 100644 --- a/tests/unit/completion/test_completionmodel.py +++ b/tests/unit/completion/test_completionmodel.py @@ -37,7 +37,7 @@ def test_first_last_item(counts): model = completionmodel.CompletionModel() for c in counts: cat = mock.Mock(spec=['layoutChanged']) - cat.rowCount = mock.Mock(return_value=c) + cat.rowCount = mock.Mock(return_value=c, spec=[]) model.add_category(cat) nonempty = [i for i, rowCount in enumerate(counts) if rowCount > 0] if not nonempty: @@ -61,7 +61,7 @@ def test_count(counts): model = completionmodel.CompletionModel() for c in counts: cat = mock.Mock(spec=['rowCount', 'layoutChanged']) - cat.rowCount = mock.Mock(return_value=c) + cat.rowCount = mock.Mock(return_value=c, spec=[]) model.add_category(cat) assert model.count() == sum(counts)