Skip to content

Commit 6c1d90a

Browse files
committed
update frame test to validate adding new column at any column index
1 parent 0d7406b commit 6c1d90a

1 file changed

Lines changed: 9 additions & 0 deletions

File tree

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

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -144,6 +144,15 @@ describe("DataFrame", function () {
144144
assert.deepEqual(df["val_count"].values, [1, 2, 3, 4]);
145145
assert.deepEqual(df["val_sum"].values, [20.3, 30.456, 40.90, 90.1]);
146146
});
147+
it("Add new array values to DataFrame works", function () {
148+
let data = { alpha: ["A", "B", "C", "D"], val_count: [1, 2, 3, 4], val_sum: [20.3, 30.456, 40.90, 90.1] };
149+
let df = new DataFrame(data);
150+
const newdf = df.addColumn("new_column", ["a", "b", "c", "d"], {atIndex: 0}) as DataFrame;
151+
assert.deepEqual(newdf["new_column"].values, ["a", "b", "c", "d"]);
152+
assert.deepEqual(newdf.columns, ["new_column", "alpha", "val_count", "val_sum"]);
153+
assert.deepEqual(newdf.dtypes, ["string", "int32", "float32", "string"]);
154+
assert.deepEqual(newdf.index, [0, 1, 2, 3]);
155+
});
147156
})
148157

149158
describe("drop", function () {

0 commit comments

Comments
 (0)