- Add license to test files

- Style transfer test
- remove .test file in posenet

Author: cvalenzuela

src/ImageClassifier/index_test.js

@@ -1,4 +1,7 @@
-/* eslint new-cap: 0 */
+// Copyright (c) 2018 ml5
+//
+// This software is released under the MIT License.
+// https://opensource.org/licenses/MIT
 
 const { tf, imageClassifier } = ml5;
 

src/LSTM/index_test.js

@@ -1,3 +1,8 @@
+// Copyright (c) 2018 ml5
+//
+// This software is released under the MIT License.
+// https://opensource.org/licenses/MIT
+
 const { LSTMGenerator } = ml5;
 
 const LSTM_MODEL_URL = 'https://raw.githubusercontent.com/ml5js/ml5-data-and-models/master/models/lstm/woolf/';

src/PoseNet/index.test.js

@@ -26,7 +26,7 @@ class StyleTransfer extends Video {
     this.epsilonScalar = tf.scalar(1e-3);
     this.video = null;
     this.ready = callCallback(this.load(model), callback);
-    this.then = this.ready.then;
+    // this.then = this.ready.then;
   }
 
   async load(model) {

src/StyleTransfer/index.js

@@ -3,7 +3,39 @@
 // This software is released under the MIT License.
 // https://opensource.org/licenses/MIT
 
-// Just a test
-// test('true should be true', () => {
-//   expect(true).toBe(true);
-// });
+
+const { styleTransfer } = ml5;
+
+const STYLE_TRANSFER_MODEL = 'https://github.com/ml5js/ml5-data-and-models/raw/master/models/style-transfer/matta/';
+const STYLE_TRANSFER_DEFAULTS = {
+  size: 200,
+};
+
+describe('styleTransfer', () => {
+  let style;
+
+  async function getRobin() {
+    const img = new Image();
+    img.crossOrigin = '';
+    img.src = 'https://ml5js.org/docs/assets/img/bird.jpg';
+    await new Promise((resolve) => { img.onload = resolve; });
+    return img;
+  }
+
+  beforeEach(async () => {
+    jasmine.DEFAULT_TIMEOUT_INTERVAL = 100000;
+    style = await styleTransfer(STYLE_TRANSFER_MODEL);
+  });
+
+  it('instantiates styleTransfer', () => {
+    expect(style.size).toBe(STYLE_TRANSFER_DEFAULTS.size);
+  });
+
+  it('styles an image', async () => {
+    // Same as with LSTM. There's an issue with the checkpoint loader
+    // const image = await getRobin();
+    // const transfer = await style.transfer(image);
+    // console.log(transfer);
+    // expect(transfer).toBe('bird');
+  });
+});

src/StyleTransfer/index_test.js

@@ -1,3 +1,8 @@
+// Copyright (c) 2018 ml5
+//
+// This software is released under the MIT License.
+// https://opensource.org/licenses/MIT
+
 /* eslint no-loop-func: 0 */
 const { word2vec } = ml5;
 

src/Word2vec/index_test.js

@@ -1,4 +1,7 @@
-/* eslint new-cap: 0 */
+// Copyright (c) 2018 ml5
+//
+// This software is released under the MIT License.
+// https://opensource.org/licenses/MIT
 
 const { YOLO } = ml5;
 

src/YOLO/index_test.js

@@ -1,3 +1,8 @@
+// Copyright (c) 2018 ml5
+//
+// This software is released under the MIT License.
+// https://opensource.org/licenses/MIT
+
 import * as tf from '@tensorflow/tfjs';
 
 const MANIFEST_FILE = 'manifest.json';