add additional tests

Author: ernestoarbitrio

src/cr/cube/matrix/subtotals.py

@@ -408,7 +408,7 @@ def subtotal_rows(cls, base_values, counts, default_insertions, dimensions):
     @lazyproperty
     def _column_subtotals(self):
         """Sequence of _Subtotal object for each subtotal in columns-dimension."""
-        return self._dimensions[-1].subtotals
+        return self._dimensions[1].subtotals
 
     def _multiple_subtrahends_or_addends(self, subtotal):
         """Returns true if the subtotal has multiple addend or subtrahend terms."""
@@ -459,7 +459,7 @@ def _subtotal_columns(self):
 
     def _subtotal_column(self, subtotal, default):
         """Return (n_rows,) ndarray of values for `subtotal` column."""
-        if self._dimensions[-1].dimension_type == DT.CAT_DATE and any(
+        if self._dimensions[1].dimension_type == DT.CAT_DATE and any(
             subtotal.subtrahend_idxs
         ):
             if self._multiple_subtrahends_or_addends(subtotal):

tests/integration/test_headers_and_subtotals.py

@@ -2764,6 +2764,43 @@ def it_provide_residual_test_stats_including_hs(self):
 class DescribeIntegrated_SubtotalDifferences:
     """TDD driver(s) for Subtotal Difference insertions."""
 
+    def it_computes_diff_for_cat_date_on_a_3D_cube(self):
+        cube = Cube(
+            CR.CAT_X_MR_X_CAT_DATE,
+            transforms={
+                "columns_dimension": {
+                    "insertions": [
+                        {
+                            "function": "subtotal",
+                            "args": [1],
+                            "kwargs": {"positive": [1], "negative": [2]},
+                            "anchor": "top",
+                            "name": "diff1",
+                        },
+                        {
+                            "function": "subtotal",
+                            "args": [1],
+                            "kwargs": {"positive": [2, 3]},
+                            "anchor": "top",
+                            "name": "sub1",
+                        },
+                    ]
+                }
+            },
+        )
+        slice_ = cube.partitions[0]
+        assert len(cube.dimensions) == 3
+        # first 2 cols are the insertions
+        assert slice_.row_proportions[:, :2] == pytest.approx(
+            np.array(
+                [
+                    [-0.26136364, 0.77272727],
+                    [0.29411765, 0.47058824],
+                    [-0.20338983, 0.74576271],
+                ]
+            )
+        )
+
     def it_computes_diff_for_cat_date(self):
         strand = Cube(
             CR.CAT_DATE,