replaced unittest assertions pytest assertions (#26539)

This commit is contained in:
Aarif
2021-02-18 19:13:12 +05:00
committed by GitHub
parent 139a0f6ee1
commit 874436b720
8 changed files with 30 additions and 29 deletions

View File

@@ -92,15 +92,15 @@ class MockSignalHandlerMixin(object):
mock_handler = Mock(spec=handler)
mock_signal.connect(mock_handler)
yield
self.assertTrue(mock_handler.called)
assert mock_handler.called
mock_args, mock_kwargs = mock_handler.call_args
if 'exclude_args' in kwargs:
for key in kwargs['exclude_args']:
self.assertIn(key, mock_kwargs)
assert key in mock_kwargs
del mock_kwargs[key]
del kwargs['exclude_args']
self.assertEqual(mock_args, args)
self.assertEqual(mock_kwargs, dict(kwargs, signal=mock_signal))
assert mock_args == args
assert mock_kwargs == dict(kwargs, signal=mock_signal)
@contextmanager