superset/superset-frontend/packages/superset-ui-core/test/math-expression/index.test.ts
Yongjie Zhao cb97e37a92
refactor: import value name from root of superset-ui/core (#17947)
* chart

* index and number-format

* char-composition and color

* connection

* dimension and math-expression

* models

* query

* time-format

* residual modules

* change config of jest and eslint

config

* wip

update package.json

* fix lint

* WithLegend import from superset-ui/core
2022-01-08 09:16:24 +08:00

100 lines
4.1 KiB
TypeScript

/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
import { evalExpression, isValidExpression } from '@superset-ui/core';
test('evalExpression evaluates constants correctly', () => {
expect(evalExpression('0', 10)).toEqual(0);
expect(evalExpression('0.123456', 0)).toEqual(0.123456);
expect(evalExpression('789', 100)).toEqual(789);
});
test('evalExpression evaluates infinities correctly', () => {
const formula = 'x/0';
expect(evalExpression(formula, 1)).toEqual(Infinity);
expect(evalExpression(formula, -1)).toEqual(-Infinity);
});
test('evalExpression evaluates powers correctly', () => {
const formula = '2^(x/2)*100';
expect(evalExpression(formula, 0)).toEqual(100);
expect(evalExpression(formula, 1)).toEqual(141.4213562373095);
expect(evalExpression(formula, 2)).toEqual(200);
});
test('evalExpression ignores whitespace and variables on left hand side of equals sign', () => {
expect(evalExpression('y=x+1', 1)).toEqual(2);
expect(evalExpression('y = x - 1', 1)).toEqual(0);
});
test('evalExpression evaluates custom operators correctly', () => {
const equalsExpression = 'x == 10';
expect(evalExpression(equalsExpression, 5)).toEqual(0);
expect(evalExpression(equalsExpression, 10)).toEqual(1);
expect(evalExpression(equalsExpression, 10.1)).toEqual(0);
const closedRange = '(x > 0) and (x < 10)';
expect(evalExpression(closedRange, 0)).toEqual(0);
expect(evalExpression(closedRange, 5)).toEqual(1);
expect(evalExpression(closedRange, 10)).toEqual(0);
const openRange = '(x >= 0) and (x <= 10)';
expect(evalExpression(openRange, -0.1)).toEqual(0);
expect(evalExpression(openRange, 0)).toEqual(1);
expect(evalExpression(openRange, 5)).toEqual(1);
expect(evalExpression(openRange, 10)).toEqual(1);
expect(evalExpression(openRange, 10.1)).toEqual(0);
const orRange = '(x < 0) or (x > 10)';
expect(evalExpression(orRange, -0.1)).toEqual(1);
expect(evalExpression(orRange, 0)).toEqual(0);
expect(evalExpression(orRange, 5)).toEqual(0);
expect(evalExpression(orRange, 10)).toEqual(0);
expect(evalExpression(orRange, 10.1)).toEqual(1);
// other less used operators
expect(evalExpression('5 & x', 3)).toEqual(1);
expect(evalExpression('5 | x', 3)).toEqual(7);
expect(evalExpression('5 xor x', 2)).toEqual(7);
// complex combinations
const complexExpression =
'20.51*(x<1577836800000)+20.2((x<15805152000000)&(x>=1577836800000))';
expect(evalExpression(complexExpression, 0)).toEqual(20.51);
expect(evalExpression(complexExpression, 1000)).toEqual(20.51);
expect(evalExpression(complexExpression, 1577836800000)).toEqual(20.2);
expect(evalExpression(complexExpression, 15805151999999)).toEqual(20.2);
expect(evalExpression(complexExpression, 15805152000000)).toEqual(0);
expect(evalExpression(complexExpression, 15805159000000)).toEqual(0);
});
test('isValidExpression correctly identifies invalid formulas', () => {
expect(isValidExpression('foobar')).toEqual(false);
expect(isValidExpression('x+')).toEqual(false);
expect(isValidExpression('z+1')).toEqual(false);
});
test('isValidExpression correctly identifies valid formulas', () => {
expect(isValidExpression('x')).toEqual(true);
expect(isValidExpression('x+1')).toEqual(true);
expect(isValidExpression('y=x-1')).toEqual(true);
expect(isValidExpression('y = x - 1')).toEqual(true);
expect(isValidExpression('y = (x < 100 and x > 0) * 100')).toEqual(true);
});