Skip to content

Commit 5e26647

Browse files
committed
Updating Series.replace() and DataFrame.replace() tests to expect error when oldValue is NaN
1 parent 2db945a commit 5e26647

File tree

2 files changed

+6
-9
lines changed

2 files changed

+6
-9
lines changed

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

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -2163,12 +2163,10 @@ describe("DataFrame", function () {
21632163
assert.deepEqual(df_rep.values, expected);
21642164
});
21652165

2166-
it("Replace oldValue supports falsy numbers (NaN)", function () {
2166+
it("Replace oldValue does not support NaN", function () {
21672167
const data1 = [[NaN, 19, 84, NaN], [65, NaN, NaN, 37]];
21682168
const df = new DataFrame(data1);
2169-
const expected = [[1, 19, 84, 1], [65, 1, 1, 37]];
2170-
const df_rep = df.replace(NaN, 1) as DataFrame;
2171-
assert.deepEqual(df_rep.values, expected);
2169+
assert.throws(() => df.replace(NaN, 1), Error, "Params Error: Param 'oldValue' does not support NaN. Use DataFrame.fillNa() instead.");
21722170
});
21732171

21742172
it("Replace oldValue supports falsy strings", function () {
@@ -2218,7 +2216,7 @@ describe("DataFrame", function () {
22182216
const df_rep = df.replace(true, false) as DataFrame;
22192217
assert.deepEqual(df_rep.values, expected);
22202218
});
2221-
2219+
22222220
});
22232221

22242222
describe("sum", function () {

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

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1188,12 +1188,11 @@ describe("Series Functions", () => {
11881188
assert.deepEqual(dfRep.values, expected);
11891189
});
11901190

1191-
it("Replace oldValue supports falsy numbers (NaN)", function () {
1191+
it("Replace oldValue does not support NaN", function () {
11921192
const data1 = [NaN, 45, 56, 25, 23, 20, NaN];
11931193
const sf = new Series(data1);
1194-
const expected = [1, 45, 56, 25, 23, 20, 1];
1195-
const dfRep = sf.replace(NaN, 1)
1196-
assert.deepEqual(dfRep.values, expected);
1194+
assert.throws(() => sf.replace(NaN, 1), Error, "Params Error: Param 'oldValue' does not support NaN. Use Series.fillNa() instead.");
1195+
11971196
});
11981197

11991198
it("Replace oldValue supports falsy strings", function () {

0 commit comments

Comments
 (0)