Test unsetting plotly-customdata via animation

Author: rreusser

src/traces/scatter/plot.js

@@ -428,9 +428,14 @@ function plotOne(gd, idx, plotinfo, cdscatter, cdscatterAll, element, transition
         }
 
         join.each(function(d) {
-            var sel = transition(d3.select(this));
+            var el = d3.select(this);
+            var sel = transition(el);
             Drawing.translatePoint(d, sel, xa, ya);
             Drawing.singlePointStyle(d, sel, trace);
+
+            if(trace.customdata) {
+                el.classed('plotly-customdata', d.data !== null && d.data !== undefined);
+            }
         });
 
         if(hasTransition) {

src/traces/scatter/style.js

@@ -30,12 +30,6 @@ module.exports = function style(gd) {
 
             pts.call(Drawing.pointStyle, trace);
 
-            if(trace.customdata) {
-                pts.each(function(cd) {
-                    d3.select(this).classed('plotly-customdata', cd.data !== null && cd.data !== undefined);
-                });
-            }
-
             el.selectAll('text')
                 .call(Drawing.textPointStyle, trace);
         });

test/jasmine/tests/scatter_test.js

@@ -358,6 +358,17 @@ describe('end-to-end scatter tests', function() {
             points.each(function(cd, i) {
                 expect(d3.select(this).classed('plotly-customdata')).toBe(expected[i]);
             });
+
+            return Plotly.animate(gd, [{
+                data: [{customdata: []}]
+            }], {frame: {redraw: false, duration: 0}});
+        }).then(function() {
+            var points = d3.selectAll('g.scatterlayer').selectAll('.point');
+
+            points.each(function() {
+                expect(d3.select(this).classed('plotly-customdata')).toBe(false);
+            });
+
         }).catch(fail).then(done);
     });
 });