Don't allow MapMutation.update() calls after the mutation is finished

Author: 1st1

immutables/_map.c

@@ -3880,6 +3880,10 @@ mapmut_py_update(MapMutationObject *self, PyObject *args, PyObject *kwds)
         return NULL;
     }
 
+    if (mapmut_check_finalized(self)) {
+        return NULL;
+    }
+
     if (arg != NULL) {
         if (map_update_inplace(self->m_mutid, (BaseMapObject *)self, arg)) {
             return NULL;

immutables/map.py

@@ -719,6 +719,9 @@ def __contains__(self, key):
             return True
 
     def update(self, col=None, **kw):
+        if self.__mutid == 0:
+            raise ValueError('mutation {!r} has been finished'.format(self))
+
         it = None
         if col is not None:
             if hasattr(col, 'items'):

tests/test_map.py

@@ -1148,6 +1148,9 @@ def test_map_mut_12(self):
         with self.assertRaisesRegex(ValueError, 'has been finished'):
             mm['a'] = 'b'
 
+        with self.assertRaisesRegex(ValueError, 'has been finished'):
+            mm.update(a='b')
+
     def test_map_mut_13(self):
         key1 = HashKey(123, 'aaa')
         key2 = HashKey(123, 'aaa')