Python: add attribute to MagicMock