From 783af7a56150901f9d7eb09fb24d8ef0df7d60b3 Mon Sep 17 00:00:00 2001
From: jos <wjosdejong@gmail.com>
Date: Mon, 9 Feb 2015 15:28:19 +0100
Subject: [PATCH] Changed unit tests to use the compiled vis.js

---
 lib/DataSet.js        | 4 +---
 test/DataSet.test.js  | 8 +++++---
 test/DataView.test.js | 8 +++++---
 3 files changed, 11 insertions(+), 9 deletions(-)

diff --git a/lib/DataSet.js b/lib/DataSet.js
index 975f40de..d8a256e2 100644
--- a/lib/DataSet.js
+++ b/lib/DataSet.js
@@ -153,9 +153,7 @@ DataSet.prototype.subscribe = DataSet.prototype.on;
 DataSet.prototype.off = function(event, callback) {
   var subscribers = this._subscribers[event];
   if (subscribers) {
-    this._subscribers[event] = subscribers.filter(function (listener) {
-      return (listener.callback != callback);
-    });
+    this._subscribers[event] = subscribers.filter(listener => listener.callback != callback);
   }
 };
 
diff --git a/test/DataSet.test.js b/test/DataSet.test.js
index 0def1526..23344a02 100644
--- a/test/DataSet.test.js
+++ b/test/DataSet.test.js
@@ -1,7 +1,9 @@
 var assert = require('assert');
-var moment = require('moment');
-var DataSet = require('../lib/DataSet');
-var Queue = require('../lib/Queue');
+var vis = require('../dist/vis');
+var moment = vis.moment;
+var DataSet = vis.DataSet;
+var Queue = vis.Queue;
+// TODO: test the source code immediately, but this is ES6
 
 var now = new Date();
 
diff --git a/test/DataView.test.js b/test/DataView.test.js
index 43110ba2..a7c8bc31 100644
--- a/test/DataView.test.js
+++ b/test/DataView.test.js
@@ -1,7 +1,9 @@
 var assert = require('assert');
-var moment = require('moment');
-var DataSet = require('../lib/DataSet');
-var DataView = require('../lib/DataView');
+var vis = require('../dist/vis');
+var moment = vis.moment;
+var DataSet = vis.DataSet;
+var DataView = vis.DataView;
+// TODO: test the source code immediately, but this is ES6
 
 // TODO: improve DataView tests, split up in one test per function
 describe('DataView', function () {