Add new test for query method

Author: risenW

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

@@ -2690,6 +2690,16 @@ describe("DataFrame", function () {
       assert.deepEqual(query_df.index, []);
     });
 
+    it("Confirms that column names are not changed", function () {
+
+      let data = [ [ 1, 2, 3 ], [ 4, 5, 6 ], [ 20, 30, 40 ], [ 39, 89, 78 ] ];
+      let cols = [ "A", "B", "C" ];
+      let df = new dfd.DataFrame(data, { columns: cols });
+      let df_query = df.query( df["B"].ge(5));
+      assert.deepEqual(df_query.index, [ 1, 2, 3 ]);
+      assert.deepEqual(df_query.columns, [ "A", "B", "C" ]);
+    });
+
   });
 
   describe("cTypes", function () {

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

@@ -2701,6 +2701,16 @@ describe("DataFrame", function () {
             assert.deepEqual(query_df.index, []);
         });
 
+        it("Confirms that column names are not changed", function () {
+
+            let data = [ [ 1, 2, 3 ], [ 4, 5, 6 ], [ 20, 30, 40 ], [ 39, 89, 78 ] ];
+            let cols = [ "A", "B", "C" ];
+            let df = new DataFrame(data, { columns: cols });
+            let df_query = df.query( df["B"].ge(5));
+            assert.deepEqual(df_query.index, [ 1, 2, 3 ]);
+            assert.deepEqual(df_query.columns, [ "A", "B", "C" ]);
+          });
+
     });
 
     describe("cTypes", function () {