def _upgrade(self): call_ugprades([self.upgrade_func])
def test_upgrade_dict(self): upgrade_func = Mock(side_effect=Exception) call_ugprades([upgrade_func]) self.assertEqual(1, upgrade_func.call_count)