-
Notifications
You must be signed in to change notification settings - Fork 0
/
test.js
55 lines (51 loc) · 1001 Bytes
/
test.js
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
const skipgram = require('./index');
const test = require('ava');
const text = '0 1 2 3 4 5 6 7 8 9';
test('1-gram', t => {
t.deepEqual(skipgram(1)(text), [
[ '0' ],
[ '1' ],
[ '2' ],
[ '3' ],
[ '4' ],
[ '5' ],
[ '6' ],
[ '7' ],
[ '8' ],
[ '9' ]
]);
});
test('2-gram', t => {
t.deepEqual(skipgram(2)(text), [
[ '0', '1' ],
[ '1', '2' ],
[ '2', '3' ],
[ '3', '4' ],
[ '4', '5' ],
[ '5', '6' ],
[ '6', '7' ],
[ '7', '8' ],
[ '8', '9' ]
]);
});
test('4-gram', t => {
t.deepEqual(skipgram(4)(text), [
[ '0', '1', '2', '3' ],
[ '1', '2', '3', '4' ],
[ '2', '3', '4', '5' ],
[ '3', '4', '5', '6' ],
[ '4', '5', '6', '7' ],
[ '5', '6', '7', '8' ],
[ '6', '7', '8', '9' ]
]);
});
test('3,1-gram', t => {
t.deepEqual(skipgram(3, 1)(text), [
[ '0', '2', '4' ],
[ '1', '3', '5' ],
[ '2', '4', '6' ],
[ '3', '5', '7' ],
[ '4', '6', '8' ],
[ '5', '7', '9' ]
]);
});