Newer
Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
import { Test, TestingModule } from '@nestjs/testing';
import { RawResult } from 'src/engine/models/result/raw-result.model';
import { AppModule } from '../../../../../main/app.module';
import { ENGINE_SERVICE } from '../../../../engine.constants';
import { IEngineService } from '../../../../engine.interfaces';
import { ExperimentCreateInput } from '../../../../models/experiment/input/experiment-create.input';
import {
createExperiment,
generateNumber,
TEST_PATHOLOGIES,
TIMEOUT_DURATION_SECONDS,
waitForResult,
} from '../../interfaces/test-utilities';
jest.setTimeout(1000 * TIMEOUT_DURATION_SECONDS);
describe('ExaremeService', () => {
let exaremeService: IEngineService;
beforeEach(async () => {
const moduleRef: TestingModule = await Test.createTestingModule({
imports: [AppModule],
}).compile();
exaremeService = await moduleRef.resolve<IEngineService>(ENGINE_SERVICE);
});
const modelSlug = `naivebayes-${generateNumber()}`;
const algorithmId = 'NAIVE_BAYES';
const input: ExperimentCreateInput = {
name: modelSlug,
variables: ['alzheimerbroadcategory'],
coVariables: ['righthippocampus', 'lefthippocampus'],
datasets: TEST_PATHOLOGIES.dementia.datasets
.filter((d) => d.code !== 'fake_longitudinal')
.map((d) => d.code),
domain: TEST_PATHOLOGIES.dementia.code,
algorithm: {
id: algorithmId,
type: 'string',
parameters: [
{
id: 'alpha',
value: ['0.1'],
},
{
id: 'k',
value: ['10'],
},
],
},
filter: '',
};
describe('Integration Test for experiment API', () => {
it(`create ${algorithmId}`, async () => {
const experiment = await createExperiment(input, exaremeService);
expect(experiment).toBeTruthy();
expect(experiment?.status).toStrictEqual('pending');
expect(experiment?.id).toBeTruthy();
const experimentResult = await waitForResult(
experiment?.id ?? '',
exaremeService,
);
expect(experimentResult).toBeTruthy();
expect(experimentResult.status).toStrictEqual('success');
expect(experimentResult.results.length).toBeGreaterThanOrEqual(5);
const data = experimentResult.results[0] as RawResult;
expect(data.rawdata['data']['precision'][0]).toBeCloseTo(0.517, 3);
});
});
});