Fix nniManager unit test (#515)

src/nni_manager/core/test/nnimanager.test.ts

@@ -57,15 +57,15 @@ describe('Unit test for nnimanager', function () {
         trainingServicePlatform: 'local',
         searchSpace: '{"x":1}',
         tuner: {
-            className: 'EvolutionTuner',
+            className: 'TPE',
             classArgs: {
                 optimize_mode: 'maximize'
             },
             checkpointDir: '',
-            gpuNum: 1
+            gpuNum: 0
         },
         assessor: {
-            className: 'MedianstopAssessor',
+            className: 'Medianstop',
             checkpointDir: '',
             gpuNum: 1
         }
@@ -74,9 +74,12 @@ describe('Unit test for nnimanager', function () {
     before(async () => {
         await initContainer();
         nniManager = component.get(Manager);
+        const expId: string = await nniManager.startExperiment(experimentParams);
+        assert(expId);
     });
 
-    after(() => {
+    after(async () => {
+        await nniManager.stopExperiment();
         cleanupUnitTest();
     })
 
@@ -84,15 +87,6 @@ describe('Unit test for nnimanager', function () {
         //TODO: add resume experiment unit test
     })
 
-    it('test startExperiment', () => {
-
-        return nniManager.startExperiment(experimentParams).then(function (experimentId) {
-            expect(experimentId.length).to.be.equal(8);
-        }).catch(function (error) {
-            assert.fail(error);
-        })
-    })
-
     it('test listTrialJobs', () => {
         //FIXME: not implemented
         //return nniManager.listTrialJobs().then(function (trialJobDetails) {
@@ -147,7 +141,6 @@ describe('Unit test for nnimanager', function () {
         })
     })
 
-
     it('test addCustomizedTrialJob', () => {
         return nniManager.addCustomizedTrialJob('hyperParams').then(() => {