[node/merge][l]: add test for merge

Author: steveoni

src/danfojs-node/package.json

@@ -29,7 +29,7 @@
     "xlsx": "^0.17.2"
   },
   "scripts": {
-    "test": "nyc mocha --require ts-node/register test/**/*.test.ts",
+    "test": "nyc mocha --require ts-node/register test/**/merge.test.ts",
     "test:clean": "yarn build:clean && yarn test",
     "dev": "nodemon",
     "build": "tsc",

src/danfojs-node/test/transformers/merge.test.ts

@@ -0,0 +1,105 @@
+import { assert } from "chai";
+import { DataFrame, merge } from "../../dist/danfojs-node/src";
+
+
+describe("Merge", function () {
+
+  it("test outer merge", function () {
+    let data = [ [ 'K0', 'k0', 'A0', 'B0' ], [ 'k0', 'K1', 'A1', 'B1' ],
+      [ 'K1', 'K0', 'A2', 'B2' ], [ 'K2', 'K2', 'A3', 'B3' ] ];
+
+    let data2 = [ [ 'K0', 'k0', 'C0', 'D0' ], [ 'K1', 'K0', 'C1', 'D1' ],
+      [ 'K1', 'K0', 'C2', 'D2' ], [ 'K2', 'K0', 'C3', 'D3' ] ];
+
+    let colum1 = [ 'Key1', 'Key2', 'A', 'B' ];
+    let colum2 = [ 'Key1', 'Key2', 'A', 'D' ];
+
+    let df1 = new DataFrame(data, { columns: colum1 });
+    let df2 = new DataFrame(data2, { columns: colum2 });
+    let merge_df = merge({ left: df1, right: df2, on: [ "Key1", "Key2" ], how: "outer" });
+
+    let output_data = [
+      [ 'K0', 'k0', 'A0', 'B0', 'C0', 'D0' ],
+      [ 'k0', 'K1', 'A1', 'B1', NaN, NaN ],
+      [ 'K1', 'K0', 'A2', 'B2', 'C1', 'D1' ],
+      [ 'K1', 'K0', 'A2', 'B2', 'C2', 'D2' ],
+      [ 'K2', 'K2', 'A3', 'B3', NaN, NaN ],
+      [ 'K2', 'K0', NaN, NaN, 'C3', 'D3' ]
+    ];
+
+    assert.deepEqual(merge_df.values, output_data);
+  });
+
+  it("test inner merge", function () {
+    let data = [ [ 'K0', 'k0', 'A0', 'B0' ], [ 'k0', 'K1', 'A1', 'B1' ],
+      [ 'K1', 'K0', 'A2', 'B2' ], [ 'K2', 'K2', 'A3', 'B3' ] ];
+
+    let data2 = [ [ 'K0', 'k0', 'C0', 'D0' ], [ 'K1', 'K0', 'C1', 'D1' ],
+      [ 'K1', 'K0', 'C2', 'D2' ], [ 'K2', 'K0', 'C3', 'D3' ] ];
+
+    let colum1 = [ 'Key1', 'Key2', 'A', 'B' ];
+    let colum2 = [ 'Key1', 'Key2', 'A', 'D' ];
+
+    let df1 = new DataFrame(data, { columns: colum1 });
+    let df2 = new DataFrame(data2, { columns: colum2 });
+    let merge_df = merge({ left: df1, right: df2, on: [ "Key1", "Key2" ], how: "inner" });
+
+    let output_data = [
+      [ 'K0', 'k0', 'A0', 'B0', 'C0', 'D0' ],
+      [ 'K1', 'K0', 'A2', 'B2', 'C1', 'D1' ],
+      [ 'K1', 'K0', 'A2', 'B2', 'C2', 'D2' ]
+    ];
+
+    assert.deepEqual(merge_df.values, output_data);
+  });
+
+  it("test left merge", function () {
+    let data = [ [ 'K0', 'k0', 'A0', 'B0' ], [ 'k0', 'K1', 'A1', 'B1' ],
+      [ 'K1', 'K0', 'A2', 'B2' ], [ 'K2', 'K2', 'A3', 'B3' ] ];
+
+    let data2 = [ [ 'K0', 'k0', 'C0', 'D0' ], [ 'K1', 'K0', 'C1', 'D1' ],
+      [ 'K1', 'K0', 'C2', 'D2' ], [ 'K2', 'K0', 'C3', 'D3' ] ];
+
+    let colum1 = [ 'Key1', 'Key2', 'A', 'B' ];
+    let colum2 = [ 'Key1', 'Key2', 'A', 'D' ];
+
+    let df1 = new DataFrame(data, { columns: colum1 });
+    let df2 = new DataFrame(data2, { columns: colum2 });
+    let merge_df = merge({ left: df1, right: df2, on: [ "Key1", "Key2" ], how: "left" });
+
+    let output_data = [
+      [ 'K0', 'k0', 'A0', 'B0', 'C0', 'D0' ],
+      [ 'k0', 'K1', 'A1', 'B1', NaN, NaN ],
+      [ 'K1', 'K0', 'A2', 'B2', 'C1', 'D1' ],
+      [ 'K1', 'K0', 'A2', 'B2', 'C2', 'D2' ],
+      [ 'K2', 'K2', 'A3', 'B3', NaN, NaN ]
+    ];
+
+    assert.deepEqual(merge_df.values, output_data);
+  });
+
+  it("test right merge", function () {
+    let data = [ [ 'K0', 'k0', 'A0', 'B0' ], [ 'k0', 'K1', 'A1', 'B1' ],
+      [ 'K1', 'K0', 'A2', 'B2' ], [ 'K2', 'K2', 'A3', 'B3' ] ];
+
+    let data2 = [ [ 'K0', 'k0', 'C0', 'D0' ], [ 'K1', 'K0', 'C1', 'D1' ],
+      [ 'K1', 'K0', 'C2', 'D2' ], [ 'K2', 'K0', 'C3', 'D3' ] ];
+
+    let colum1 = [ 'Key1', 'Key2', 'A', 'B' ];
+    let colum2 = [ 'Key1', 'Key2', 'A', 'D' ];
+
+    let df1 = new DataFrame(data, { columns: colum1 });
+    let df2 = new DataFrame(data2, { columns: colum2 });
+    let merge_df = merge({ left: df1, right: df2, on: [ "Key1", "Key2" ], how: "right" });
+
+    let output_data = [
+      [ 'K0', 'k0', 'A0', 'B0', 'C0', 'D0' ],
+      [ 'K1', 'K0', 'A2', 'B2', 'C1', 'D1' ],
+      [ 'K1', 'K0', 'A2', 'B2', 'C2', 'D2' ],
+      [ 'K2', 'K0', NaN, NaN, 'C3', 'D3' ]
+    ];
+
+    assert.deepEqual(merge_df.values, output_data);
+  });
+})
+