X-Git-Url: https://adrianiainlam.tk/git/?a=blobdiff_plain;ds=sidebyside;f=auto_tests%2Ftests%2Fselection.js;h=bb8e8522a9ea56c089c9ea2d9381aef4897b5edf;hb=d97bf726ccc5645e60171f86b134fb29ac31b846;hp=02f1227a7c793162f3805a01574d99fbb5d64431;hpb=1b7afc93886e2927b6f320e8e1ed4dd6bd9ab53b;p=dygraphs.git

diff --git a/auto_tests/tests/selection.js b/auto_tests/tests/selection.js
index 02f1227..bb8e852 100644
--- a/auto_tests/tests/selection.js
+++ b/auto_tests/tests/selection.js
@@ -6,11 +6,12 @@
  * @author danvk@google.com (Dan Vanderkam)
  */
 
+import Dygraph from '../../src/dygraph';
+import DefaultHandler from '../../src/datahandler/default';
+
 describe("selection", function() {
 
-beforeEach(function() {
-  document.body.innerHTML = "<div id='graph'></div>";
-});
+cleanupAfterEach();
 
 it('testSetGetSelection', function() {
   var graph = document.getElementById("graph");
@@ -51,7 +52,7 @@ it('testSetGetSelectionDense', function() {
 
 it('testSetGetSelectionMissingPoints', function() {
   var dataHandler = function() {};
-  dataHandler.prototype = new Dygraph.DataHandlers.DefaultHandler();
+  dataHandler.prototype = new DefaultHandler();
   dataHandler.prototype.seriesToPoints = function(series, setName, boundaryIdStart) {
     var val = null;
     if (setName == 'A') {