From 55d59992c6695e2ba4093b1d76905f3263dfba3e Mon Sep 17 00:00:00 2001 From: Emma Foley Date: Mon, 12 Mar 2018 15:07:59 +0000 Subject: assert[Greater,Equal] -> assert_{,not_}called assertEqual(mock_xxx.call_count, 1) -> mock_xxx.assert_called_once assertEqual(mock_xxx.call_count, 0) -> mock_xxx.assert_not_called assertGreater(mock.call_count, 0) -> mock.assert_called() assertGreaterEqual(mock.call_count, 1) -> mock.assert_called() JIRA: YARDSTICK-1069 Change-Id: I890084d120c8e78304e169e2a0e5d30011a41525 Signed-off-by: Emma Foley --- yardstick/tests/unit/network_services/test_yang_model.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) (limited to 'yardstick/tests/unit/network_services/test_yang_model.py') diff --git a/yardstick/tests/unit/network_services/test_yang_model.py b/yardstick/tests/unit/network_services/test_yang_model.py index a7eb36b8a..cbeb3a1f2 100644 --- a/yardstick/tests/unit/network_services/test_yang_model.py +++ b/yardstick/tests/unit/network_services/test_yang_model.py @@ -119,11 +119,11 @@ class YangModelTestCase(unittest.TestCase): y._rules = None self.assertIsNone(y.get_rules()) - self.assertEqual(read_mock.call_count, 1) - self.assertEqual(get_mock.call_count, 1) + read_mock.assert_called_once() + get_mock.assert_called_once() # True value should prevent calling read and get y._rules = 999 self.assertEqual(y.get_rules(), 999) - self.assertEqual(read_mock.call_count, 1) - self.assertEqual(get_mock.call_count, 1) + read_mock.assert_called_once() + get_mock.assert_called_once() -- cgit 1.2.3-korg