anydict.py

from collections import UserDict
from unittest.mock import ANY


class AnyDict(UserDict):
    """
    Returns a dict that will use unittest.mock.ANY
    as a value for keys that present in dict it is
    going to be compared with for equality (or not equality),
    but not in dict itself. It will also ensure that comparing
    dict is just a dict, not OrderedDict or other subtype.
    """

    def _fill_with_any(self, other):
        missing_keys = set(other) - set(self.data)
        self.data = {**self.data, **{key: ANY for key in missing_keys}}

    def __eq__(self, other):
        self._fill_with_any(other)
        return super().__eq__(dict(other))

    def __ne__(self, other):
        self._fill_with_any(other)
        return super().__ne__(dict(other))