improve the unit testing coverage

chen0040 · May 26, 2017 · a2fe60d · a2fe60d
1 parent 03a2746
commit a2fe60d
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 0 deletions.
diff --git a/test/linear-regression-spec.js b/test/linear-regression-spec.js
@@ -55,6 +55,11 @@ describe("Test linear regression", function() {
           expect(intercept).to.above(0.0); 
       });
 
+      it("can transform multiple rows of data", function(){
+         var Y_predicted = regression.transform(data); 
+          expect(Y_predicted.length).to.equal(data.length);
+      });
+
   });
 });
 
diff --git a/test/logistic-regression-spec.js b/test/logistic-regression-spec.js
@@ -44,5 +44,10 @@ describe("Test logistic regression", function(){
        it('should have a cost of lower than 0.5', function(){
           expect(result.cost).to.below(0.6); 
        });
+
+        it("can transform multiple rows of data", function(){
+             var Y_predicted = logistic.transform(testingData); 
+              expect(Y_predicted.length).to.equal(testingData.length);
+        });
     });
 });
diff --git a/test/multi-class-logistic-spec.js b/test/multi-class-logistic-spec.js
@@ -45,5 +45,10 @@ describe("Test multi-class classification using logistic regression", function()
            }
 
        });
+
+        it("can transform multiple rows of data", function(){
+            var Y_predicted = classifier.transform(testingData); 
+            expect(Y_predicted.length).to.equal(testingData.length);
+        });
     });
 });