def __exit__(self, exc_type, exc_value, tb):
if exc_type is None: # <- Values are None when no exeption was raised.
if self.msg:
msg = self.msg
else:
msg = getattr(self.function, '__name__', str(self.function))
exc = AssertionError('No differences found: ' + str(msg))
exc.__cause__ = None
raise exc
if not issubclass(exc_type, DataError):
raise exc_value # If not DataError, re-raise without changes.
diffs = exc_value.differences
rejected_kwds, accepted_kwds = self._partition_kwds(diffs, **self.kwds)
rejected_func = self.function(accepted_kwds) # <- Apply function!
not_allowed = itertools.chain(rejected_kwds, rejected_func)
not_allowed = list(not_allowed)
if not_allowed:
msg = [self.msg, getattr(exc_value, 'msg')]
msg = ': '.join(x for x in msg if x)
exc = DataError(msg, not_allowed)
exc.__cause__ = None # <- Suppress context using verbose
raise exc # alternative to support older Python
# versions--see PEP 415 (same as
# effect as "raise ... from None").
return True # <- Suppress original exception.
def __init__(self, data, fieldnames=None):
first_value, iterator = iterpeek(data)
if isinstance(first_value, dict):
if not fieldnames:
fieldnames = list(first_value.keys())
super(DataSource, self).__init__(iterator, fieldnames)
else:
if fieldnames:
iterator = itertools.chain([fieldnames], iterator)
super(DataSource, self).__init__(iterator)