chore: Unit tests

Author: robertleeplummerjr

lib/agents/determinist-pg.test.ts

@@ -40,7 +40,7 @@ describe("DeterministPG", () => {
         outputSize: 7,
         activation: "sigmoid",
       });
-      expect(new DeterministPG(agent.toJSON())).toMatchObject(agent);
+      expect(new DeterministPG(agent.toJSON()).toJSON()).toMatchObject(agent.toJSON());
     });
   });
 });

lib/agents/dqn-agent.test.ts

@@ -49,7 +49,7 @@ describe("DQNAgent", () => {
         outputSize: 10,
         activation: "sigmoid",
       });
-      expect(new DQNAgent(agent.toJSON())).toMatchObject(agent);
+      expect(new DQNAgent(agent.toJSON()).toJSON()).toMatchObject(agent.toJSON());
     });
   });
 });

lib/lstm.test.ts

@@ -33,7 +33,7 @@ describe("LSTM", () => {
           Wch: new Mat(2, 2),
           bc: new Mat(2, 1),
         }],
-        Whd: new Mat(3, 0),
+        Whd: new Mat(3, 2),
         bd: new Mat(3, 1),
       });
     });

lib/net.test.ts

@@ -79,14 +79,14 @@ describe("Net", () => {
           W2: net.weights[1].toJSON(),
           b2: net.biases[1].toJSON(),
         };
-        expect(Net.fromJSON(json)).toEqual(net);
+        expect(Net.fromJSON(json).toJSON()).toEqual(net.toJSON());
       });
     });
     describe("when using standard json", () => {
       it("calls fromJSON() provides equivalent net", () => {
         const net = new Net(1, [2] ,3);
         const json = net.toJSON();
-        expect(Net.fromJSON(json)).toEqual(net);
+        expect(Net.fromJSON(json).toJSON()).toEqual(net.toJSON());
       });
     });
     describe("when using unknown json", () => {

lib/rand-mat.test.ts

@@ -6,8 +6,6 @@ describe("RandMat", () => {
       const mat = new RandMat(2, 2, 2, 0.08);
       expect(mat.n).toEqual(2);
       expect(mat.d).toEqual(2);
-      expect(mat.mu).toEqual(2);
-      expect(mat.std).toEqual(0.08);
       expect(mat.w).not.toEqual(new Float64Array([0,0,0,0]));
     });
   });

lib/rand-mat.ts

@@ -5,13 +5,11 @@ export class RandMat extends Mat {
   constructor(
     n: number,
     d: number,
-    public mu: number,
-    public std: number
+    private mu: number,
+    private std: number
   ) {
     super(n, d);
     this.fillRandn(mu, std);
     //fillRand(this,-std,std); // kind of :P
   }
-
-  //TODO: mu and std from JSON?
 }