diff options
Diffstat (limited to 'yardstick/tests/unit/network_services/test_yang_model.py')
-rw-r--r-- | yardstick/tests/unit/network_services/test_yang_model.py | 8 |
1 files changed, 4 insertions, 4 deletions
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() |