implement LArray.rename which does not modify the axis name inplace

gdementen · gdementen · commit ed19cfea46d7 · 2015-01-28T17:03:33.000+01:00
diff --git a/larray/core.py b/larray/core.py
@@ -870,6 +870,12 @@ def axes_rename(self, **kwargs):
         self.axes = AxisCollection(axes)
         return self
 
+    def rename(self, axis, newname):
+        axis = self.get_axis(axis)
+        axes = [Axis(newname, a.labels) if a is axis else a
+                for a in self.axes]
+        return LArray(self, axes)
+
     def full_key(self, key):
         """
         Returns a full nd-key from a key in any of the following forms:
diff --git a/larray/tests/test_la.py b/larray/tests/test_la.py
@@ -518,6 +518,18 @@ def test_zeros_like(self):
         self.assertEqual(la.shape, (116, 44, 2, 15))
         self._assert_equal_raw(la, np.zeros((116, 44, 2, 15)))
 
+    def test_rename(self):
+        la = self.larray
+        new = la.rename('sex', 'gender')
+        # old array axes names not modified
+        self.assertEqual(la.axes_names, ['age', 'geo', 'sex', 'lipro'])
+        self.assertEqual(new.axes_names, ['age', 'geo', 'gender', 'lipro'])
+
+        new = la.rename(self.sex, 'gender')
+        # old array axes names not modified
+        self.assertEqual(la.axes_names, ['age', 'geo', 'sex', 'lipro'])
+        self.assertEqual(new.axes_names, ['age', 'geo', 'gender', 'lipro'])
+
     def test_info(self):
         expected = """\
 116 x 44 x 2 x 15
