diff --git a/tests/utils/fuzzy_equal.py b/tests/utils/fuzzy_equal.py index 0832e675..d087b853 100644 --- a/tests/utils/fuzzy_equal.py +++ b/tests/utils/fuzzy_equal.py @@ -21,7 +21,7 @@ def wrapper(self, actual, expected, context=None): self.stack.append(context) is_pb_savable = any(hasattr(actual, s) for s in ['__getstate__', '__getinitargs__']) - kind = pb.save if is_pb_savable else actual.__class__ + kind = type(pb.save) if is_pb_savable else actual.__class__ dispatcher.dispatch(kind)(self, actual, expected) if context is not None and self.stack: @@ -148,7 +148,7 @@ def _(self, actual, expected): for key in actual: self._assert(actual[key], expected[key], context="['{}']".format(key)) - @_assert.register(pb.save) + @_assert.register(type(pb.save)) def _(self, actual, expected): specials = [s for s in ['__getstate__', '__getinitargs__'] if hasattr(actual, s)] for s in specials: