better test comments

gdementen · gdementen · commit a83ace1ac915 · 2015-01-28T12:06:28.000+01:00
diff --git a/larray/tests/test_la.py b/larray/tests/test_la.py
@@ -597,7 +597,10 @@ def test_getitem(self):
         # la[[1, 5, 9], age['1,5,9']]
         self.assertRaises(ValueError, la.__getitem__, ([1, 5], age['1,5']))
 
-    def test_setitem(self):
+    def test_setitem_larray(self):
+        """
+        tests LArray.__setitem__(key, value) where value is an LArray
+        """
         age, geo, sex, lipro = self.larray.axes
 
         # 1) using a ValueGroup key
@@ -611,7 +614,7 @@ def test_setitem(self):
         raw[[1, 5, 9]] = raw[[1, 5, 9]] + 25.0
         self._assert_equal_raw(la, raw)
 
-        # b) same size but a different shape (extra length-1 axis)
+        # b) value has an extra length-1 axis
         la = self.larray.copy()
         raw = self.array.copy()
 
@@ -624,16 +627,17 @@ def test_setitem(self):
         raw[[1, 5, 9]] = raw[[1, 5, 9]] + 26.0
         self._assert_equal_raw(la, raw)
 
-        # dimension of length 1
+        # c) value has the same axes than target but one has length 1
         la = self.larray.copy()
         raw = self.array.copy()
         raw[[1, 5, 9]] = np.sum(raw[[1, 5, 9]], axis=1, keepdims=True)
         la[ages1_5_9] = la[ages1_5_9].sum(geo=(geo.all(),))
         self._assert_equal_raw(la, raw)
 
-        # c) missing dimension
+        # d) value has a missing dimension
         la = self.larray.copy()
         la[ages1_5_9] = la[ages1_5_9].sum(geo)
+        # we use "raw" from previous test
         self._assert_equal_raw(la, raw)
 
         # 2) using a string key
