From 23f3ef71bbe788798f4f604807264af113e1c5cd Mon Sep 17 00:00:00 2001
From: Gonzalo Bellver <gonzalo.bellver@gmail.com>
Date: Fri, 27 Jan 2012 11:52:34 -0300
Subject: [PATCH] Added test for BalancedTestSuite

---
 mindplot/src/test/javascript/static/layout.html  |  3 +++
 .../javascript/static/test/BalancedTestSuite.js  | 16 +++++++++++++++-
 2 files changed, 18 insertions(+), 1 deletion(-)

diff --git a/mindplot/src/test/javascript/static/layout.html b/mindplot/src/test/javascript/static/layout.html
index 2536fae5..2fd62416 100644
--- a/mindplot/src/test/javascript/static/layout.html
+++ b/mindplot/src/test/javascript/static/layout.html
@@ -115,6 +115,9 @@
     <div id="testBalancedPredict3"></div>
     <div id="testBalancedPredict4"></div>
     <div id="testBalancedPredict5"></div>
+    
+    <h3>testBalancedSingleNodePredict</h3>
+    <div id="testBalancedSingleNodePredict1"></div>
 </div>
 
 <div id="symmetricTest" style="display: none;">
diff --git a/mindplot/src/test/javascript/static/test/BalancedTestSuite.js b/mindplot/src/test/javascript/static/test/BalancedTestSuite.js
index 0c020362..8fa82d9e 100644
--- a/mindplot/src/test/javascript/static/test/BalancedTestSuite.js
+++ b/mindplot/src/test/javascript/static/test/BalancedTestSuite.js
@@ -23,6 +23,7 @@ mindplot.layout.BalancedTestSuite = new Class({
 
         this.testBalanced();
         this.testBalancedPredict();
+        this.testBalancedSingleNodePredict();
     },
 
     testBalanced: function() {
@@ -151,7 +152,7 @@ mindplot.layout.BalancedTestSuite = new Class({
     },
 
     testBalancedPredict: function() {
-        console.log("testBalancedPredict");
+        console.log("testBalancedPredict:");
         var position = {x:0, y:0};
         var manager = new mindplot.layout.LayoutManager(0, mindplot.layout.TestSuite.ROOT_NODE_SIZE);
 
@@ -273,5 +274,18 @@ mindplot.layout.BalancedTestSuite = new Class({
         $assert(prediction5a.order == prediction5b.order, "Both predictions should be the same");
 
         console.log("OK!\n\n");
+    },
+
+    testBalancedSingleNodePredict: function() {
+        console.log("testBalancedSingleNodePredict:");
+        var position = {x:0, y:0};
+        var manager = new mindplot.layout.LayoutManager(0, mindplot.layout.TestSuite.ROOT_NODE_SIZE);
+
+        manager.addNode(1, mindplot.layout.TestSuite.NODE_SIZE, position).connectNode(0,1,0);
+        manager.layout();
+        var graph = manager.plot("testBalancedSingleNodePredict1", {width:800, height:400});
+
+        var prediction1 = manager.predict(0, {x:50, y:50});
+        this._plotPrediction(graph, prediction1);
     }
 });
\ No newline at end of file