assert_equal(): Delegate to assert_dict_equal() if applicable

#14

Author: srittau

NEWS.rst

@@ -9,7 +9,8 @@ API Additions
 Improvements
 ------------
 
-* `fail()` is now marked with `NoReturn` in type stub.
+* ``assert_equal()``: Use ``assert_dict_equal()`` if applicable.
+* ``fail()`` is now marked with ``NoReturn`` in type stub.
 
 Bug Fixes
 ---------

asserts/__init__.py

@@ -165,7 +165,9 @@ def assert_equal(first, second, msg_fmt="{msg}"):
     * second - the second argument
     """
 
-    if not first == second:
+    if isinstance(first, dict) and isinstance(second, dict):
+        assert_dict_equal(first, second, msg_fmt)
+    elif not first == second:
         msg = "{!r} != {!r}".format(first, second)
         fail(msg_fmt.format(msg=msg, first=first, second=second))
 

test_asserts.py

@@ -187,6 +187,10 @@ def test_assert_equal__not_equal__custom_message(self):
         with _assert_raises_assertion("'string' != 55;'string';55"):
             assert_equal("string", 55, "{msg};{first!r};{second!r}")
 
+    def test_assert_equal__dict(self):
+        with _assert_raises_assertion("key 'foo' missing from right dict"):
+            assert_equal({"foo": 5}, {})
+
     # assert_not_equal()
 
     def test_assert_not_equal__not_equal(self):