toimage_test: make sure that all graph divs are deleted

Author: etpinard

test/jasmine/tests/toimage_test.js

@@ -2,20 +2,27 @@
 //  once established and confirmed?
 
 var Plotly = require('@lib/index');
+
+var d3 = require('d3');
 var createGraphDiv = require('../assets/create_graph_div');
-var destroyGraphDiv = require('../assets/destroy_graph_div');
 var subplotMock = require('@mocks/multiple_subplots.json');
 
 
 describe('Plotly.toImage', function() {
     'use strict';
+
     var gd;
 
     beforeEach(function() {
         gd = createGraphDiv();
     });
 
-    afterEach(destroyGraphDiv);
+    afterEach(function() {
+
+        // make sure ALL graph divs are deleted,
+        // even the ones generated by Plotly.toImage
+        d3.selectAll('.js-plotly-plot').remove();
+    });
 
     it('should be attached to Plotly', function() {
         expect(Plotly.toImage).toBeDefined();

test/jasmine/tests/transforms_test.js

@@ -776,6 +776,7 @@ describe('multiple traces with transforms:', function() {
 
 function assertDims(dims) {
     var traces = d3.selectAll('.trace');
+
     expect(traces.size())
         .toEqual(dims.length, 'to have correct number of traces');