diff --git a/test/__init__.py b/test/__init__.py index 69759ffa4..349f30a96 100644 --- a/test/__init__.py +++ b/test/__init__.py @@ -583,10 +583,10 @@ def sanitize_reply(reply): class PyMongoTestCase(unittest.TestCase): def assertEqualCommand(self, expected, actual, msg=None): - self.assertEqual(expected, sanitize_cmd(actual), msg) + self.assertEqual(sanitize_cmd(expected), sanitize_cmd(actual), msg) def assertEqualReply(self, expected, actual, msg=None): - self.assertEqual(expected, sanitize_reply(actual), msg) + self.assertEqual(sanitize_reply(expected), sanitize_reply(actual), msg) class IntegrationTest(PyMongoTestCase): diff --git a/test/test_database.py b/test/test_database.py index c70329916..cb297eb7f 100644 --- a/test/test_database.py +++ b/test/test_database.py @@ -417,7 +417,7 @@ class TestDatabase(IntegrationTest): # getLastError includes "connectionId" in recent # server versions, getPrevError does not. error.pop("connectionId", None) - self.assertEqual(error, prev_error) + self.assertEqualReply(error, prev_error) db.test.find_one() self.assertEqual(None, db.error())