【fix】UT review by qiwei

songyumeng · songyumeng · commit 5e416be50602 · 2021-04-29T15:31:02.000+08:00
diff --git a/test/common/thirdparty/ai/BinaryClassificationSpec.js b/test/common/thirdparty/ai/BinaryClassificationSpec.js
@@ -11,6 +11,9 @@ describe('binary classification', () => {
         testImage.src = image.src;
         window.document.body.appendChild(testImage);
     });
+    afterAll(() => {
+        window.document.body.removeChild(testImage);
+    });
     it('predict', (done) => {
         var params = {
             modelUrl: 'http://fakeurl/model.json',
diff --git a/test/common/thirdparty/ai/LandcoverClassificationSpec.js b/test/common/thirdparty/ai/LandcoverClassificationSpec.js
@@ -11,6 +11,9 @@ describe('landcover classification', () => {
         testImage.src = image.src;
         window.document.body.appendChild(testImage);
     });
+    afterAll(() => {
+        window.document.body.removeChild(testImage);
+    });
     it('predict', (done) => {
         var params = {
             modelUrl: 'http://fakeurl/model.json',
diff --git a/test/common/thirdparty/ai/ObjectDetectionSpec.js b/test/common/thirdparty/ai/ObjectDetectionSpec.js
@@ -11,6 +11,9 @@ describe('object detection', () => {
         testImage.src = image.src;
         window.document.body.appendChild(testImage);
     });
+    afterAll(() => {
+        window.document.body.removeChild(testImage);
+    });
     it('predict', (done) => {
         var params = {
             modelUrl: 'http://fakeurl/model.json',
diff --git a/test/common/thirdparty/ai/WebMachineLearningSpec.js b/test/common/thirdparty/ai/WebMachineLearningSpec.js
@@ -1,5 +1,6 @@
 import { BinaryClassification } from '../../../../src/common/thirdparty/ai/BinaryClassification';
 import { WebMachineLearning } from '../../../../src/common/thirdparty/ai/WebMachineLearning';
+import image from '../../../resources/img/baiduTileTest.png';
 import * as tfconv from '@tensorflow/tfjs-converter';
 import * as tfcore from '@tensorflow/tfjs-core';
 
@@ -8,13 +9,15 @@ describe('WebMachineLearning', () => {
     beforeAll(() => {
         testImage = window.document.createElement('img');
         testImage.id = 'demo';
-        testImage.src = testImage;
+        testImage.src = image.src;
         window.document.body.appendChild(testImage);
     });
-
+    afterAll(() => {
+        window.document.body.removeChild(testImage);
+    });
     it('predict', (done) => {
         var params = {
-            modelUrl: 'http://192.168.11.148:8080/model.json',
+            modelUrl: 'ttp://fakeurl/model.json',
             image: window.document.querySelector('#demo')
         };
         var binaryClassification = new BinaryClassification(params);
