FIX: fixed most remaining CheckedSession warnings stacklevel problems

there is still one problematic test for pandas_hdf engine but there seems to be something very fishy going on

Author: gdementen

larray/core/checked.py

@@ -478,6 +478,10 @@ def _check_key_value(self, name: str, value: Any, skip_allow_mutation: bool, ski
                 warnings.warn(f"'{name}' is not declared in '{self.__class__.__name__}'", stacklevel=stacklevel + 1)
             return value
 
+        def _update_from_iterable(self, it):
+            for k, v in it:
+                self.__setitem__(k, v, stacklevel=3)
+
         def __setitem__(self, key, value, skip_allow_mutation=False, skip_validation=False, stacklevel=1):
             if key != 'meta':
                 value = self._check_key_value(key, value, skip_allow_mutation, skip_validation,

larray/core/session.py

@@ -139,10 +139,8 @@ def add(self, *args, **kwargs):
         >>> s.names
         ['arr1', 'arr2', 'arr3', 'x', 'y']
         """
-        for arg in args:
-            self[arg.name] = arg
-        for k, v in kwargs.items():
-            self[k] = v
+        self._update_from_iterable((arg.name, arg) for arg in args)
+        self._update_from_iterable(kwargs.items())
 
     def update(self, other=None, **kwargs):
         r"""
@@ -243,16 +241,13 @@ def update(self, other=None, **kwargs):
         if other is None:
             pass
         elif hasattr(other, 'items'):
-            for k, v in other.items():
-                self[k] = v
+            self._update_from_iterable(other.items())
         elif isinstance(other, Iterable):
-            for k, v in other:
-                self[k] = v
+            self._update_from_iterable(other)
         else:
             raise ValueError(f"Expected Session, dict-like or iterable object for 'other' argument. "
                              f"Got {type(other).__name__}.")
-        for k, v in kwargs.items():
-            self[k] = v
+        self._update_from_iterable(kwargs.items())
 
     def _ipython_key_completions_(self):
         return list(self.keys())
@@ -446,10 +441,13 @@ def load(self, fname, names=None, engine='auto', display=False, **kwargs):
         else:
             handler = handler_cls(fname)
         metadata, objects = handler.read(names, display=display, **kwargs)
-        for k, v in objects.items():
-            self[k] = v
+        self._update_from_iterable(objects.items())
         self.meta = metadata
 
+    def _update_from_iterable(self, it):
+        for k, v in it:
+            self[k] = v
+
     def save(self, fname, names=None, engine='auto', overwrite=True, display=False, **kwargs):
         r"""
         Dumps objects from the current session to a file, or several .csv files.