fix #832 : implemented CheckedSession, CheckedParameters and CheckedArray classes (based on preliminary code written by gdementen)

Author: alixdamman

README.rst

@@ -129,6 +129,12 @@ For plotting
 - `matplotlib <http://matplotlib.org/>`__:
   required for plotting.
 
+Miscellaneous
+~~~~~~~~~~~~~
+
+- `pydantic <https://github.com/samuelcolvin/pydantic>`__:
+  required to use `CheckedSession`.
+
 .. _start-documentation:
 
 Documentation

doc/source/api.rst

@@ -795,7 +795,6 @@ Modifying
 
    Session.add
    Session.update
-   Session.get
    Session.apply
    Session.transpose
 
@@ -821,6 +820,30 @@ Load/Save
    Session.to_hdf
    Session.to_pickle
 
+CheckedArray
+============
+
+.. autosummary::
+   :toctree: _generated/
+
+   CheckedArray
+
+CheckedSession
+==============
+
+.. autosummary::
+   :toctree: _generated/
+
+   CheckedSession
+
+CheckedParameters
+=================
+
+.. autosummary::
+   :toctree: _generated/
+
+   CheckedParameters
+
 .. _api-editor:
 
 Editor

environment.yml

@@ -11,5 +11,6 @@ dependencies:
   - xlsxwriter
   - pytest>=3.5
   - flake8
+  - pydantic<=1.5
   - pip:
     - pytest-flake8

larray/__init__.py

@@ -8,6 +8,7 @@
                                eye, all, any, sum, prod, cumsum, cumprod, min, max, mean, ptp, var,
                                std, median, percentile, stack, zip_array_values, zip_array_items)
 from larray.core.session import Session, local_arrays, global_arrays, arrays
+from larray.core.checked import CheckedArray, CheckedSession, CheckedParameters
 from larray.core.constants import nan, inf, pi, e, euler_gamma
 from larray.core.metadata import Metadata
 from larray.core.ufuncs import wrap_elementwise_array_func, maximum, minimum, where
@@ -55,6 +56,8 @@
     'median', 'percentile', 'stack', 'zip_array_values', 'zip_array_items',
     # session
     'Session', 'local_arrays', 'global_arrays', 'arrays',
+    # constrained
+    'CheckedArray', 'CheckedSession', 'CheckedParameters',
     # constants
     'nan', 'inf', 'pi', 'e', 'euler_gamma',
     # metadata

larray/core/axis.py

@@ -839,7 +839,7 @@ def __getitem__(self, key):
         -----
         key is label-based (slice and fancy indexing are supported)
         """
-        # if isinstance(key, basestring):
+        # if isinstance(key, str):
         #     key = to_keys(key)
 
         def isscalar(k):
@@ -862,7 +862,7 @@ def isscalar(k):
             and key.name in self
         ):
             return LGroup(key.name, None, self)
-        # elif isinstance(key, basestring) and key in self:
+        # elif isinstance(key, str) and key in self:
             # TODO: this is an awful workaround to avoid the "processing" of string keys which exist as is in the axis
             #       (probably because the string was used in an aggregate function to create the label)
             # res = LGroup(slice(None), None, self)