make the assertions python 3.5 friendly

canonical · Jan 11, 2024 · 7eef520 · 7eef520
1 parent 3f65a12
commit 7eef520
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/providers/base/tests/test_cpuid.py b/providers/base/tests/test_cpuid.py
@@ -40,7 +40,10 @@ def test_hygon_dhyana_plus(self, cpuid_mock, co_mock, print_mock):
         cpuid_mock.return_value = call_mock
         co_mock.return_value = ""
         main()
-        self.assertIn("Hygon Dhyana Plus", print_mock.mock_calls[1].args[0])
+        expected_msg = "CPUID: {} which appears to be a {} processor".format(
+            "0x900f22", "Hygon Dhyana Plus"
+        )
+        print_mock.assert_called_with(expected_msg)
 
     @patch("subprocess.check_output")
     @patch("cpuid.CPUID")