1 files changed, 18 insertions, 11 deletions
diff --git a/test/unit_tests/layer.test.js b/test/unit_tests/layer.test.js
index 37402d3..d7dcdb8 100644
--- a/test/unit_tests/layer.test.js
+++ b/test/unit_tests/layer.test.js
@@ -1,15 +1,22 @@
-console.log('Hello from layer.test.js');
+const BinaryConverter = require('../../src/binaryConverter');
+const Layer = require('../../src/layer');
+const math = require('mathjs');
 
-let testLayer = new Layer(2, 3, 'relu');
-let testConv = new BinaryConverter(2);
+describe('Test Layer module', () => {
 
-console.log(testLayer);
-console.log(testConv.randomInput());
+    let testLayer = new Layer(2, 3, 'relu');
+    let testConv = new BinaryConverter(2);
+    
+    testConv.randomInput()
+    testLayer.forwardPropogation(testConv.inputActivation);
+    
+    testLayer = new Layer(2, 3, 'sigmoid');
+    testLayer.forwardPropogation(testConv.inputActivation);
+    
+    testLayer = new Layer(2, 3, 'identity');
+    testLayer.forwardPropogation(testConv.inputActivation);
+        
+    test.todo('Layer Tests');
 
-console.log(testLayer.forwardPropogation(testConv.inputActivation));
+});
 
-testLayer = new Layer(2, 3, 'sigmoid');
-console.log(testLayer.forwardPropogation(testConv.inputActivation));
-
-testLayer = new Layer(2, 3, 'identity');
-console.log(testLayer.forwardPropogation(testConv.inputActivation));