fix(formula): fix codecov

dream-num · Sep 9, 2024 · b29574d · b29574d
1 parent 9182d9e
commit b29574d
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 4 deletions.
diff --git a/packages/engine-formula/src/functions/lookup/drop/index.ts b/packages/engine-formula/src/functions/lookup/drop/index.ts
@@ -132,10 +132,6 @@ export class Drop extends BaseFunction {
     }
 
     private _getResultArray(array: BaseValueObject, rows: number, columns: number, arrayRowCount: number, arrayColumnCount: number): BaseValueObject {
-        if (!array.isArray()) {
-            return array;
-        }
-
         const rowParam = rows >= 0 ? [rows, arrayRowCount] : [0, arrayRowCount + rows];
         const columnParam = columns >= 0 ? [columns, arrayColumnCount] : [0, arrayColumnCount + columns];
 

diff --git a/packages/engine-formula/src/functions/lookup/expand/__tests__/index.spec.ts b/packages/engine-formula/src/functions/lookup/expand/__tests__/index.spec.ts
@@ -90,6 +90,9 @@ describe('Test expand function', () => {
                 [ErrorType.VALUE, ErrorType.VALUE],
                 [ErrorType.VALUE, ErrorType.VALUE],
             ]);
+
+            const result7 = testFunction.calculate(array2, rows, columns);
+            expect(getObjectValue(result7)).toStrictEqual(1);
         });
 
         it('Rows value test', async () => {

diff --git a/packages/engine-formula/src/functions/lookup/take/__tests__/index.spec.ts b/packages/engine-formula/src/functions/lookup/take/__tests__/index.spec.ts
@@ -94,6 +94,9 @@ describe('Test take function', () => {
                 [ErrorType.VALUE, ErrorType.VALUE],
                 [ErrorType.VALUE, ErrorType.VALUE],
             ]);
+
+            const result7 = testFunction.calculate(array2, rows, columns);
+            expect(getObjectValue(result7)).toStrictEqual(1);
         });
 
         it('Rows value test', async () => {