(issue 805): anonymous and/or wildcard axes are handled correctly when using Session.save() and Session.load()

Author: alixdamman

larray/inout/pandas.py

@@ -1,5 +1,6 @@
 from __future__ import absolute_import, print_function
 
+import re
 from itertools import product
 from collections import OrderedDict
 
@@ -140,6 +141,9 @@ def from_series(s, sort_rows=False, fill_value=nan, meta=None, **kwargs):
         return LArray(s.values, Axis(s.index.values, name), meta=meta)
 
 
+_anonymous_axis_pattern = re.compile(r'\{(\d+|\??)\}\*?')
+
+
 def from_frame(df, sort_rows=False, sort_columns=False, parse_header=False, unfold_last_axis_name=False,
                fill_value=nan, meta=None, cartesian_prod=True, **kwargs):
     r"""
@@ -357,15 +361,20 @@ def _extract_labels_from_series(series):
 
 
 def _axis_to_series(key, axis, dtype=None):
-    name = '{}:{}'.format(key, axis.name)
-    return pd.Series(data=axis.labels, name=name, dtype=dtype)
+    name = '{}:{}'.format(key, str(axis))
+    labels = len(axis) if axis.iswildcard else axis.labels
+    return pd.Series(data=labels, name=name, dtype=dtype)
 
 
 def _series_to_axis(series):
     name = str(series.name)
     labels = _extract_labels_from_series(series)
     if ':' in name:
         key, axis_name = name.split(':')
+        if axis_name[-1] == '*':
+            labels = labels[0]
+        if _anonymous_axis_pattern.match(axis_name):
+            axis_name = None
     else:
         # for backward compatibility
         key = axis_name = name