fix #4232 - fill in value for generated sectors

... with the partialSum to compute the correct partition under
    `branchvalues: 'total'`. Remember, we "generate" sectors
    for "implied root" and "root of roots" scenarios.

Author: etpinard

src/traces/sunburst/calc.js

@@ -106,6 +106,7 @@ exports.calc = function(gd, trace) {
         if(impliedRoots.length === 1) {
             k = impliedRoots[0];
             cd.unshift({
+                hasImpliedRoot: true,
                 id: k,
                 pid: '',
                 label: k
@@ -153,12 +154,20 @@ exports.calc = function(gd, trace) {
                 break;
             case 'total':
                 hierarchy.each(function(d) {
-                    var v = d.data.data.v;
+                    var cdi = d.data.data;
+                    var v = cdi.v;
 
                     if(d.children) {
                         var partialSum = d.children.reduce(function(a, c) {
                             return a + c.data.data.v;
                         }, 0);
+
+                        // N.B. we must fill in `value` for generated sectors
+                        // with the partialSum to compute the correct partition
+                        if(cdi.hasImpliedRoot || cdi.hasMultipleRoots) {
+                            v = partialSum;
+                        }
+
                         if(v < partialSum) {
                             failed = true;
                             return Lib.warn([

test/jasmine/tests/sunburst_test.js

@@ -185,7 +185,7 @@ describe('Test sunburst calc:', function() {
     function extractPt(k) {
         var out = gd.calcdata.map(function(cd) {
             return cd[0].hierarchy.descendants().map(function(pt) {
-                return pt[k];
+                return Lib.nestedProperty(pt, k).get();
             });
         });
         return out.length > 1 ? out : out[0];
@@ -317,6 +317,54 @@ describe('Test sunburst calc:', function() {
         expect(extract('id')).toEqual(['true', '1', '2', '3', '4', '5', '6', '7', '8']);
         expect(extract('pid')).toEqual(['', 'true', 'true', '2', '2', 'true', 'true', '6', 'true']);
     });
+
+    it('should compute correct sector *value* for generated implied root', function() {
+        _calc([{
+            labels: [ 'A', 'B', 'b'],
+            parents: ['Root', 'Root', 'B'],
+            values: [1, 2, 1],
+            branchvalues: 'remainder'
+        }, {
+            labels: [ 'A', 'B', 'b'],
+            parents: ['Root', 'Root', 'B'],
+            values: [1, 2, 1],
+            branchvalues: 'total'
+        }]);
+
+        expect(extractPt('data.data.id')).toEqual([
+            ['Root', 'B', 'A', 'b'],
+            ['Root', 'B', 'A', 'b']
+        ]);
+        expect(extractPt('value')).toEqual([
+            [4, 3, 1, 1],
+            [3, 2, 1, 1]
+        ]);
+    });
+
+    it('should compute correct sector *value* for generated "root of roots"', function() {
+        spyOn(Lib, 'randstr').and.callFake(function() { return 'dummy'; });
+
+        _calc([{
+            labels: [ 'A', 'B', 'b'],
+            parents: ['', '', 'B'],
+            values: [1, 2, 1],
+            branchvalues: 'remainder'
+        }, {
+            labels: [ 'A', 'B', 'b'],
+            parents: ['', '', 'B'],
+            values: [1, 2, 1],
+            branchvalues: 'total'
+        }]);
+
+        expect(extractPt('data.data.id')).toEqual([
+            ['dummy', 'B', 'A', 'b'],
+            ['dummy', 'B', 'A', 'b']
+        ]);
+        expect(extractPt('value')).toEqual([
+            [4, 3, 1, 1],
+            [3, 2, 1, 1]
+        ]);
+    });
 });
 
 describe('Test sunburst hover:', function() {