fix recent tests update with unittest2pytest

2019-10-31 23:06:40 +01:00
parent d8924ed892
commit 81647d67a0
3 changed files with 13 additions and 16 deletions
--- a/tests/test_connection.py
+++ b/tests/test_connection.py
@@ -625,8 +625,8 @@ class ConnectionTest(unittest.TestCase):
            alias="conn1", host="mongodb://localhost/testing?w=1&journal=true"
        )
        conn2 = connect("testing", alias="conn2", w=1, journal=True)
-        self.assertEqual(conn1.write_concern.document, {"w": 1, "j": True})
-        self.assertEqual(conn2.write_concern.document, {"w": 1, "j": True})
+        assert conn1.write_concern.document == {"w": 1, "j": True}
+        assert conn2.write_concern.document == {"w": 1, "j": True}

    def test_connect_with_replicaset_via_uri(self):
        """Ensure connect() works when specifying a replicaSet via the