UserDict不被单元测试视为dict

Question

import unittest
from UserDict import UserDict

class MyDict(UserDict):
    def __init__(self, x):
        UserDict.__init__(self, x=x)

class Test(unittest.TestCase):
    def test_dict(self):
        m = MyDict(42)
        assert {'x': 42} == m # this passes
        self.assertDictEqual({'x': 42}, m) # failure at here

if __name__ == '__main__':
    unittest.main()

我有

AssertionError: Second argument is not a dictionary

我应该使用内置的dict作为基类，而不是UserDict吗？

Answer 1

问题是assertDictEqual()首先检查两个参数是否为dict实例：

def assertDictEqual(self, d1, d2, msg=None):
    self.assertIsInstance(d1, dict, 'First argument is not a dictionary')
    self.assertIsInstance(d2, dict, 'Second argument is not a dictionary')
    ...

并且， UserDict不是dict的实例：

>>> m = UserDict(x=42)
>>> m
{'x': 42}
>>> isinstance(m, dict)
False

而不是直接使用UserDict类，使用包含真实字典的data属性：

self.assertDictEqual({'x': 42}, m.data)

或者，正如其他人已经建议的那样，只需使用常规字典。

Answer 2

问题是UserDict实际上不是一个dict对象，因为很久以前当你无法从内置的dict类型继承时它就被创建了。 根据文件 ：

[UserDict]已被很大程度上取代了直接从dict子类化的能力......

所以我可能只是从dict继承并完成它; 我没有看到UserDict提供的任何功能。

---

请注意，我也个人使用UserDict ^* ，因为模块名称违反了PEP8 ，这很烦人。

---

^*直到八分钟前我还不知道。

Answer 3

来自Python unittess源码

def assertDictEqual(self, d1, d2, msg=None):
   self.assertIsInstance(d1, dict, 'First argument is not a dictionary')
   self.assertIsInstance(d2, dict, 'Second argument is not a dictionary')

这两个参数必须是dict的实例。

不幸的是，UserDict实例不是dict的实例

assert isinstance(m, dict) # this is False

正如文档https://docs.python.org/2/library/userdict.html#module-UserDict所说，您可以使用m.data返回真正的字典。

IterableUserDict.data

 A real dictionary used to store the contents of the UserDict class.