DecisionTree#clone API

oguzeroglu · Sep 24, 2020 · 32d3aa7 · 32d3aa7
1 parent 369661e
commit 32d3aa7
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/js/decision-tree/DecisionTree.js b/js/decision-tree/DecisionTree.js
@@ -19,4 +19,8 @@ DecisionTree.prototype.makeDecision = function(knowledge){
   return currentNode;
 }
 
+DecisionTree.prototype.clone = function(){
+  return new DecisionTree(this._rootNode.clone());
+}
+
 export { DecisionTree };
diff --git a/test/decision-tree/DecisionTreeTest.js b/test/decision-tree/DecisionTreeTest.js
@@ -25,6 +25,20 @@ describe("DecisionTree", function(){
     expect(catchedErr.message).to.eql("Root node must be an instance of Ego.Decision")
   });
 
+  it("should clone", function(){
+
+    var rootDecision = new Ego.Decision("isStuffHappening", Ego.InformationTypes.TYPE_BOOLEAN, new Ego.IsTrue());
+
+    var decisionTree = new Ego.DecisionTree(rootDecision);
+
+    var cloned = decisionTree.clone();
+
+    expect(decisionTree).to.eql(cloned);
+    expect(decisionTree === cloned).to.eql(false);
+    expect(decisionTree._rootNode).to.eql(cloned._rootNode);
+    expect(decisionTree._rootNode === cloned._rootNode).to.eql(false);
+  });
+
   describe("should make decision", function(){
 
     it("root node only", function(){