Add tests to CharRNN (#307) (#320)

* add tests to CharRNN

* test(CharRNN): add tests to CharRNN

added descriptive tests to ensure CharRNN behaves like its example

* remove dist

Author: joeyklee

src/CharRNN/index_test.js

@@ -3,26 +3,58 @@
 // This software is released under the MIT License.
 // https://opensource.org/licenses/MIT
 
-const { LSTMGenerator } = ml5;
+const { charRNN } = ml5;
 
-const LSTM_MODEL_URL = 'https://raw.githubusercontent.com/ml5js/ml5-data-and-models/master/models/lstm/woolf/';
-const LSTM_MODEL_DEFAULTS = {
+const RNN_MODEL_URL = 'https://raw.githubusercontent.com/ml5js/ml5-data-and-models/master/models/lstm/woolf';
+
+const RNN_MODEL_DEFAULTS = {
   cellsAmount: 2,
-  vocabSize: 90,
-  firstChar: 61,
+  vocabSize: 223
 };
 
-describe('LSTMGenerator', () => {
-  let lstm;
+const RNN_DEFAULTS = {
+  seed: 'a',
+  length: 20,
+  temperature: 0.5,
+  stateful: false
+}
+
+const RNN_OPTIONS = {
+  seed: 'the meaning of pizza is: ',
+  length: 500,
+  temperature: 0.7
+}
+
+describe('charRnn', () => {
+  let rnn;
 
   beforeAll(async () => {
-    // jasmine.DEFAULT_TIMEOUT_INTERVAL = 10000;
-    // lstm = await LSTMGenerator(LSTM_MODEL_URL);
+    jasmine.DEFAULT_TIMEOUT_INTERVAL = 30000; //set extra long interval due to issues with CharRNN generation time
+    rnn  = await charRNN(RNN_MODEL_URL, undefined);
   });
 
-  it('instantiates a lstm generator', async () => {
-    // expect(lstm.cellsAmount).toBe(LSTM_MODEL_DEFAULTS.cellsAmount);
-    // expect(lstm.vocabSize).toBe(LSTM_MODEL_DEFAULTS.vocabSize);
-    // expect(lstm.vocab[0]).toBe(LSTM_MODEL_DEFAULTS.firstChar);
+  it('instantiates an rnn with all the defaults', async () => {
+    expect(rnn.ready).toBeTruthy();
+    expect(rnn.defaults.seed).toBe(RNN_DEFAULTS.seed);
+    expect(rnn.defaults.length).toBe(RNN_DEFAULTS.length);
+    expect(rnn.defaults.temperature).toBe(RNN_DEFAULTS.temperature);
+    expect(rnn.defaults.stateful).toBe(RNN_DEFAULTS.stateful);
+  });
+
+  //  it('loads the model with all the defaults', async () => {
+  //    expect(rnn.cellsAmount).toBe(RNN_MODEL_DEFAULTS.cellsAmount);
+  //    expect(rnn.vocabSize).toBe(RNN_MODEL_DEFAULTS.vocabSize);
+  //  });
+
+  describe('generate', () => {
+    it('Should generate content that follows default options if given an empty object', async() => {
+      const result = await rnn.generate({});
+      expect(result.sample.length).toBe(20);
+    });
+
+    it('generates content that follows the set options', async() => {
+      const result = await rnn.generate(RNN_OPTIONS);
+      expect(result.sample.length).toBe(500);
+    });
   });
 });