@cloudpss/expression 0.6.0-alpha.14 → 0.6.0-alpha.16

package/tests/migrate.ts

@@ -1,7 +1,7 @@
 import { migrateMathJs } from '@cloudpss/expression/migrate';
 import type { ExpressionSource } from '../dist/expression.js';
 import { Scope } from '../dist/scope.js';
-import { Evaluator, Expression } from '../dist/index.js';
+import { Evaluator, Expression, VmError } from '../dist/index.js';
 import dedent from 'dedent';
 
 const s = new Scope({
@@ -40,8 +40,16 @@ describe('migrate', () => {
         ['sin(0)', 'sin(0)', 0],
         ['sin(f)', 'sin(f)', Math.sin(1)],
         ['f', 'f', 1],
-        ['$a', '$a', null],
-        ['$a + 1', 'matrix.add($a, 1)', 1],
+        ['o == null', 'o == nil', false],
+        ['o == undefined', 'o == nil', false],
+        ['o == undefined', 'o == nil', false],
+        ['b == false', 'b =~ 0', true],
+        ['true != b', '1 !~ b', true],
+        ['f == true', 'f =~ 1', true],
+        ['false == f', '0 =~ f', false],
+        ['0 <= f + 1 <= 2', 'f + 1 is 0..2', true],
+        ['$a', '$a', VmError],
+        ['$a + 1', 'matrix.add($a, 1)', VmError],
         ['size(a)[1]', 'len(a)', 3],
         ['size(s)[1]', 'len(chars(s))', 3],
         ['(a).length', 'len((a))', 3],
@@ -55,6 +63,8 @@ describe('migrate', () => {
         ['f >0', 'f > 0', true],
         ['f >=0', 'f >= 0', true],
         ['f <0', 'f < 0', false],
+        ['f == 1', 'f =~ 1', true],
+        ['f != 1', 'f !~ 1', false],
         ['1<f<2', '1 < f && f < 2', false],
         ['1 + 2 + f', '1 + 2 + f', 4],
         ['PI', '@pi', Math.PI],
@@ -78,6 +88,8 @@ describe('migrate', () => {
         ['- f', '-f', -1],
         ['size(f)', 'matrix.size(f)', []],
         ['size(m)', 'matrix.size(m)', [2, 2]],
+        [`[1,2,3] == f`, `[1, 2, 3]::map(fn { it =~ f })`, [true, false, false]],
+        [`$V != [1,2,3]`, `[1, 2, 3]::map(fn { $V !~ it })`, VmError],
         [
             'u.length',
             dedent`
@@ -88,7 +100,7 @@ describe('migrate', () => {
                 // - W: 符号 'u' 未定义
                 // # 原始 math.js 表达式
                 // u.length`,
-            null,
+            VmError,
         ],
         ['o["x"]', 'o.x', 1],
         ['o.x', 'o.x', 1],
@@ -129,7 +141,7 @@ describe('migrate', () => {
                 // - W: 符号 'u' 未定义
                 // # 原始 math.js 表达式
                 // a[u]`,
-            1,
+            VmError,
         ],
         [
             'u[1]',
@@ -141,7 +153,7 @@ describe('migrate', () => {
                 // - W: 符号 'u' 未定义
                 // # 原始 math.js 表达式
                 // u[1]`,
-            null,
+            VmError,
         ],
         [
             'm[1,2]',
@@ -166,11 +178,10 @@ describe('migrate', () => {
         [
             'filter(x) = x > 1; a.filter(filter)',
             dedent`
-                fn filter(x) { x > 1 }
+                fn filter(x) { matrix.entrywise(x, 1, fn (a, b) { a > b }) }
                 a::global.filter(filter)
                 // # 转换日志
                 // - W: 使用了非精确转换的语法或函数，结果可能不准确
-                // - W: '>' 不支持矩阵，计算结果可能不一致
                 // # 原始 math.js 表达式
                 // filter(x) = x > 1; a.filter(filter)`,
             [2, 3],
@@ -224,9 +235,41 @@ describe('migrate', () => {
                 `,
             [1, 4, 9],
         ],
+        ['', '', null],
+        ['#', '//', null],
+        ['#xx', '//xx', null],
+        ['(random() - 1) < 0', '(random() - 1) < 0', true],
+        [
+            're(12) + im(i)',
+            dedent`
+                /* re */(12) + /* im */(i)
+                // # 转换日志
+                // - E: 不支持复数
+                // # 原始 math.js 表达式
+                // re(12) + im(i)
+                `,
+            VmError,
+        ],
+        ['is(a, "number")', "type(a) == 'number'", false],
+        ['is(a, "string")', "type(a) == 'string'", false],
+        ['is(a, "boolean")', "type(a) == 'boolean'", false],
+        ['is(a, "null")', "type(a) == 'nil'", false],
+        ['is(a, "undefined")', "type(a) == 'nil'", false],
+        [`sum([a > 1])`, `sum([matrix.entrywise(a, 1, fn (a, b) { a > b })]::flatten())`, 2],
+        [`prod(1,2,3)`, `product(1, 2, 3)`, 6],
+        [`sum(true)`, `sum(true)`, 1],
+        [`sum(m)`, `sum(m::flatten())`, 10],
+        [`prod([1,2,3])`, `product([1, 2, 3])`, 6],
+        [`prod(f)`, `product(f)`, 1],
     ])('should migrate expression: %s', (f, t, r) => {
         const migrated = migrateMathJs(f as ExpressionSource, false, s);
         expect(migrated).toBe(t);
-        expect(e.evaluate(Expression(migrated), s)).toEqual(r);
+        if (typeof r === 'function' && r.prototype instanceof Error) {
+            expect(() => {
+                e.evaluate(Expression(migrated), s);
+            }).toThrow(r);
+        } else {
+            expect(e.evaluate(Expression(migrated), s)).toEqual(r);
+        }
     });
 });

package/tests/scope.ts

@@ -10,7 +10,7 @@ describe('Scope should work correctly', () => {
     it('can create from function', () => {
         const s = new Scope((k) => (k === 'a' ? 12 : undefined), false);
         expect(e.evaluate(Expression('a'), s)).toBe(12);
-        expect(e.evaluate(Expression('b'), s)).toBe(null);
+        expect(e.evaluate(Expression('`b` in global'), s)).toBe(false);
     });
     it('can create from object', () => {
         const s = new Scope({ a: 12, b: Expression('a + 1') }, false);
@@ -22,8 +22,8 @@ describe('Scope should work correctly', () => {
 describe('Scope should not leak dangerous objects', () => {
     it('should not access prototype fields', () => {
         const s = new Scope({}, false);
-        expect(e.evaluate(Expression('__proto__'), s)).toBe(null);
-        expect(e.evaluate(Expression('toString'), s)).toBe(null);
+        expect(e.evaluate(Expression('`__proto__` in global'), s)).toBe(false);
+        expect(e.evaluate(Expression('`toString` in global'), s)).toBe(false);
     });
     it('should access same name props', () => {
         const s = new Scope({ ['__proto__']: 12, toString: 'x' }, false);