mirrors/cpython
1
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-07-23 03:05:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 15

Issue #23581: Add matmul support to MagicMock.

Browse source 
Patch by Håkan Lövdahl.
This commit is contained in:
Berker Peksag 2015-03-12 20:42:48 +02:00
parent 4a0e14730b
commit 9bd8af788d
3 changed files with 14 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

11
Lib/unittest/test/testmock/testmagicmethods.py
View file

@ -424,5 +424,16 @@ class TestMockingMagicMethods(unittest.TestCase):
self.assertEqual(list(m), [])
def test_matmul(self):
m = MagicMock()
self.assertIsInstance(m @ 1, MagicMock)
m.__matmul__.return_value = 42
m.__rmatmul__.return_value = 666
m.__imatmul__.return_value = 24
self.assertEqual(m @ 1, 42)
self.assertEqual(1 @ m, 666)
m @= 24
self.assertEqual(m, 24)
if __name__ == '__main__':
unittest.main()