webppl
Version:
Probabilistic programming for the web
257 lines (232 loc) • 7.1 kB
JavaScript
;
var _ = require('lodash');
var path = require('path');
var webppl = require('../src/main');
var errors = require('../src/errors/errors');
var parseV8 = require('../src/errors/parsers').parseV8;
var stackTraceTestFns = {
test: function(s, k, a) {
// This simulates an exception in back-end code been throw after
// the stack has been cleared. In this setting there is no entry
// on the JS stack corresponding to a webppl call made from user
// code.
return function() {
throw new Error();
};
}
};
var testDefs = [
{ name: 'top-level',
code: 'null[0]',
stack: [{line: 1, col: 0, name: null}],
debug: true
},
{ name: 'top-level',
code: 'null[0]',
stack: [{line: 1, col: 0, name: null}],
debug: false
},
{ name: 'depth 1',
code: 'var f = function() { null[0]; };\nf();',
stack: [
{line: 1, col: 21, name: null},
{line: 2, col: 0, name: 'f'}],
debug: true
},
{ name: 'depth 1',
code: 'var f = function() { null[0]; };\nf();',
stack: [{line: 1, col: 21, name: null}],
debug: false
},
{ name: 'depth 2',
code: 'var g = function() { a; };\nvar f = function() { g(); };\nf();',
stack: [
{line: 1, col: 21, name: 'a'},
{line: 2, col: 21, name: 'g'},
{line: 3, col: 0, name: 'f'}],
debug: true
},
{ name: 'intermediate value not a function',
code: 'var f = function() {}; f()();',
stack: [{line: 1, col: 23, name: null}],
debug: true
},
{ name: 'header',
code: 'first(null);',
stack: [
{file: 'header.wppl', name: null},
{line: 1, col: 0, file: 'webppl:program', name: 'first'}],
debug: true
},
{ name: 'header',
code: 'first(null);',
stack: [
{file: 'header.wppl', name: null}],
debug: false
},
{ name: 'JS',
code: 'util.error("")',
stack: [
{file: RegExp(path.join('src', 'util.js') + '$'), webppl: false, name: null},
{line: 1, col: 5, file: 'webppl:program', name: 'error'}],
debug: true
},
{ name: 'JS',
code: 'util.error("")',
stack: [
{file: RegExp(path.join('src', 'util.js') + '$'), webppl: false, name: null},
{line: 1, col: 5, file: 'webppl:program', name: 'error'}],
debug: false
},
{ name: 'native',
code: ['var f = function() {',
' var a = {};',
' var b = {a: a};',
' _.assign(a, {b: b});',
' JSON.stringify(a);',
'};',
'f();'].join('\n'),
stack: [
{line: null, col: null, native: true, name: null},
{line: 5, col: 7, webppl: true, name: 'stringify'},
{line: 7, col: 0, webppl: true, name: 'f'}],
debug: true
},
{ name: 'native',
code: ['var f = function() {',
' var a = {};',
' var b = {a: a};',
' _.assign(a, {b: b});',
' JSON.stringify(a);',
'};',
'f();'].join('\n'),
stack: [
{line: null, col: null, native: true, name: null},
{line: 5, col: 7, webppl: true, name: 'stringify'}],
debug: false
},
{ name: 'call site not in address map',
code: ['mapData({data: [0]}, function(x) {',
' null[0];',
'})'].join('\n'),
stack: [{line: 2, col: 2, name: null}],
debug: true
},
{ name: 'before any function evaluation',
code: 'var a = b; 0',
stack: [{line: 1, col: 8, name: 'b'}],
debug: true
},
{ name: 'no webppl entry in stack trace',
code: 'util.error("")',
stack: [{file: RegExp(path.join('src', 'util.js') + '$'), webppl: false, name: null}],
debug: true,
limit: 1
},
{ name: 'top most webppl entry is not a user function',
code: ['var testFn = stackTraceTestFns.test;',
'var f = function() { return testFn(); };',
'f();'
].join('\n'),
stack: [
{file: RegExp('test-stack-trace.js$'), webppl: false, line: 16, col: 12},
{webppl: false}, {webppl: false}, {webppl: false}, // Uninteresting entries from runner.
{webppl: true, line: 3, col: 0, name: 'f'}],
debug: true
},
// The idea here is to test that the stack is as expected at the
// error which occurs after we continue from the sample statement
// for the second time.
// For this to work as expected, the current address must be saved
// after invoking the continuation and before the error.
{ name: 'after continuation',
code: ['var d = Bernoulli({p: .5});',
'var g = function() {',
' var x = sample(d);',
' assert.ok(x === false);',
' return x;',
'};',
'var f = function() {',
' return g();',
'};',
'Enumerate(function() {',
' return f();',
'});'].join('\n'),
stack: [{line: 4, col: 9, name: 'ok'},
{line: 8, col: 9, name: 'g'},
{line: 11, col: 9, name: 'f'},
{line: 10, col: 0, name: 'Enumerate'}],
debug: true
},
{ name: 'with caching transform',
code: ["Infer({method: 'incrementalMH',",
' model: function() {',
' assert.ok(false);',
'}});'].join('\n'),
stack: [{line: 3, col: 9, name: 'ok'}],
debug: true
}
];
function testEntry(entry, props, test) {
if (_.has(props, 'line')) {
test.strictEqual(entry.lineNumber, props.line, 'Unexpected line number.');
}
if (_.has(props, 'col')) {
test.strictEqual(entry.columnNumber, props.col, 'Unexpected column number.');
}
if (_.has(props, 'name')) {
test.strictEqual(entry.name, props.name, 'Unexpected name.');
}
if (_.has(props, 'webppl')) {
test.strictEqual(entry.webppl, props.webppl);
}
if (_.has(props, 'file')) {
if (props.file instanceof RegExp) {
test.ok(entry.fileName.match(props.file));
} else {
test.strictEqual(entry.fileName, props.file, 'Unexpected file name.');
}
}
}
function getStack(code, debug, limit) {
var stack, oldLimit;
try {
oldLimit = Error.stackTraceLimit;
Error.stackTraceLimit = (limit !== undefined) ? limit : oldLimit;
global.stackTraceTestFns = stackTraceTestFns;
webppl.run(code, null, {debug: debug});
} catch (e) {
stack = errors.recoverStack(e, parseV8);
} finally {
delete global.stackTraceTestFns;
Error.stackTraceLimit = oldLimit;
}
return stack;
}
function runTest(def, test) {
var stack = getStack(def.code, def.debug, def.limit);
var expectedStack = def.stack;
//console.log(JSON.stringify(stack));
//console.log(JSON.stringify(expectedStack));
if (stack.length < expectedStack.length) {
test.ok(false, 'Stack is smaller than expected.');
test.done();
return;
}
_.each(expectedStack, function(expectedEntry, i) {
testEntry(stack[i], expectedEntry, test);
});
test.done();
}
function generateTests() {
_.each(testDefs, function(def, i) {
var k = def.debug ? 'debug' : 'noDebug';
if (!exports[k]) {
exports[k] = {};
}
exports[k][def.name] = function(test) {
runTest(def, test);
};
});
}
generateTests();