Skip to content

Commit 4c38c90

Browse files
committed
test: add missing unit test for diff() with series along axis 0
1 parent f49f16c commit 4c38c90

2 files changed

Lines changed: 12 additions & 0 deletions

File tree

src/danfojs-browser/tests/core/frame.test.js

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1085,6 +1085,12 @@ describe("DataFrame", function () {
10851085
const df = new dfd.DataFrame(data);
10861086
assert.deepEqual((df.diff(sf)).values, [ [ -1, 0, 3 ], [ 9, 8, 9 ], [ 0, 0, 2 ] ]);
10871087
});
1088+
it("Return difference of a DataFrame with a Series along axis 0", function () {
1089+
const data = [ [ 0, 2, 4 ], [ 10, 10, 10 ], [ 1, 2, 3 ] ];
1090+
const sf = new dfd.Series([ 1, 2, 1 ]);
1091+
const df = new dfd.DataFrame(data);
1092+
assert.deepEqual((df.diff(sf, { axis: 0 })).values, [ [ -1, 1, 3 ], [ 8, 8, 8 ], [ 0, 1, 2 ] ]);
1093+
});
10881094
it("Return difference of a DataFrame with along axis 0 (column-wise), previous column", function () {
10891095
const data = [ [ 0, 2, 4 ], [ 10, 10, 10 ], [ 1, 2, 3 ] ];
10901096
const df = new dfd.DataFrame(data);

src/danfojs-node/test/core/frame.test.ts

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1014,6 +1014,12 @@ describe("DataFrame", function () {
10141014
const df = new DataFrame(data);
10151015
assert.deepEqual((df.diff(sf) as DataFrame).values, [[-1, 0, 3], [9, 8, 9], [0, 0, 2]]);
10161016
});
1017+
it("Return difference of a DataFrame with a Series along axis 0", function () {
1018+
const data = [[0, 2, 4], [10, 10, 10], [1, 2, 3]];
1019+
const sf = new Series([1, 2, 1]);
1020+
const df = new DataFrame(data);
1021+
assert.deepEqual((df.diff(sf, {axis: 0}) as DataFrame).values, [[-1, 1, 3], [8, 8, 8], [0, 1, 2]]);
1022+
});
10171023
it("Return difference of a DataFrame with along axis 0 (column-wise), previous column", function () {
10181024
const data = [[0, 2, 4], [10, 10, 10], [1, 2, 3]];
10191025
const df = new DataFrame(data);

0 commit comments

Comments
 (0)