Merge "ansible: disable Extra cloud image kernel stub"

[yardstick.git] / yardstick / tests / unit / benchmark / scenarios / lib / test_check_numa_info.py

diff --git a/yardstick/tests/unit/benchmark/scenarios/lib/test_check_numa_info.py b/yardstick/tests/unit/benchmark/scenarios/lib/test_check_numa_info.py
index f983f9c..270c9d3 100644 (file)
--- a/yardstick/tests/unit/benchmark/scenarios/lib/test_check_numa_info.py
+++ b/yardstick/tests/unit/benchmark/scenarios/lib/test_check_numa_info.py
@@ -19,7 +19,7 @@ class CheckNumaInfoTestCase(unittest.TestCase):
         scenario_cfg = {'info1': {}, 'info2': {}}
         obj = CheckNumaInfo(scenario_cfg, {})
         obj.run({})
-        self.assertTrue(mock_check_vm2.called)
+        mock_check_vm2.assert_called_once()
 
     def test_check_vm2_status_length_eq_1(self):
         info1 = {
@@ -74,11 +74,3 @@ class CheckNumaInfoTestCase(unittest.TestCase):
         obj = CheckNumaInfo(scenario_cfg, {})
         status = obj._check_vm2_status(info1, info2)
         self.assertFalse(status)
-
-
-def main():
-    unittest.main()
-
-
-if __name__ == '__main__':
-    main()