summaryrefslogtreecommitdiff
path: root/jstests/aggregation/bugs/server5044.js
blob: 945f31c302cdfaf36a637507ce93e63659f420fc (plain)
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
// SERVER-5044 Standard deviation expression operators

var t = db.server5044;

function test(data, popExpected, sampExpected) {
    t.drop();
    assert.writeOK(t.insert({}));  // need one document to ensure we get output

    for (var i = 0; i < data.length; i++)
        assert.writeOK(t.insert({num: data[i]}));

    var res = t.aggregate({
        $group: {
            _id: 1,
            pop: {$stdDevPop: '$num'},
            samp: {$stdDevSamp: '$num'},
        }
    }).next();

    if (popExpected === null) {
        assert.isnull(res.pop);
    } else {
        assert.close(res.pop, popExpected, '', 10 /*decimal places*/);
    }

    if (sampExpected === null) {
        assert.isnull(res.samp);
    } else {
        assert.close(res.samp, sampExpected, '', 10 /*decimal places*/);
    }
}

test([], null, null);
test([1], 0, null);
test([1, 1], 0, 0);

test(['a'], null, null);
test([1, 'a'], 0, null);
test([1, 'a', 1], 0, 0);

test([1, 2], .5, Math.sqrt(.5));
test([1, 2, 3], Math.sqrt(2 / 3), 1);

// test from http://en.wikipedia.org/wiki/Algorithms_for_calculating_variance#Example
test([4, 7, 13, 16], Math.sqrt(22.5), Math.sqrt(30));
test([1e8 + 4, 1e8 + 7, 1e8 + 13, 1e8 + 16], Math.sqrt(22.5), Math.sqrt(30));
test([1e9 + 4, 1e9 + 7, 1e9 + 13, 1e9 + 16], Math.sqrt(22.5), Math.sqrt(30));
test([1e10 + 4, 1e10 + 7, 1e10 + 13, 1e10 + 16], Math.sqrt(22.5), Math.sqrt(30));