datafolder 1.0.2 → 1.0.4

package/scripts/flights_queries.js

@@ -1,4 +1,4 @@
-const assert = require("node:assert/strict");
+import { deepEqual } from "node:assert/strict";
 
 // Return the names of all airports one can reach
 // directly (1 step) from Los Angeles International
@@ -6,94 +6,93 @@ const assert = require("node:assert/strict");
 
 function query1(qyDB) {
     console.time("query1");
-    const resultMap = {};
-    for (const x of qyDB.queryFiles("flights", { _from: "airports/LAX" })) {
-        resultMap[x.fileContent._to] = 1;
-    }
-    assert.deepEqual(Object.keys(resultMap), [
-        "airports/CVG",
-        "airports/SMF",
-        "airports/ATL",
-        "airports/MSP",
-        "airports/DFW",
-        "airports/PHL",
-        "airports/IAH",
-        "airports/TUS",
-        "airports/ORD",
-        "airports/PDX",
-        "airports/MIA",
-        "airports/SFO",
-        "airports/DEN",
-        "airports/IAD",
-        "airports/SJC",
-        "airports/PHX",
-        "airports/SEA",
-        "airports/SLC",
-        "airports/JFK",
-        "airports/SAN",
-        "airports/ONT",
-        "airports/LAS",
-        "airports/MCI",
-        "airports/MDW",
-        "airports/FLL",
-        "airports/AUS",
-        "airports/EWR",
-        "airports/CLE",
-        "airports/SBP",
-        "airports/MEM",
-        "airports/SBA",
-        "airports/CLT",
-        "airports/COS",
-        "airports/ASE",
-        "airports/ICT",
-        "airports/HOU",
-        "airports/YUM",
-        "airports/STL",
-        "airports/FAT",
-        "airports/ABQ",
-        "airports/BOI",
-        "airports/BOS",
-        "airports/LIH",
-        "airports/HNL",
-        "airports/SMX",
-        "airports/MRY",
-        "airports/DTW",
-        "airports/RNO",
-        "airports/MCO",
-        "airports/OXR",
-        "airports/OAK",
-        "airports/OGG",
-        "airports/KOA",
-        "airports/BWI",
-        "airports/CLD",
-        "airports/IND",
-        "airports/PSP",
-        "airports/ELP",
-        "airports/TPA",
-        "airports/OKC",
-        "airports/MSY",
-        "airports/SNA",
-        "airports/IYK",
-        "airports/IPL",
-        "airports/SAT",
-        "airports/MOD",
-        "airports/BNA",
-        "airports/DCA",
-        "airports/SGU",
-        "airports/BFL",
-        "airports/EUG",
-        "airports/EGE",
-        "airports/GEG",
-        "airports/BDL",
-        "airports/JAX",
-        "airports/CMH",
-        "airports/PIT",
-        "airports/RDU",
-        "airports/SJU",
-        "airports/XNA",
-        "airports/MTJ",
-        "airports/JAC",
-    ]);
+    deepEqual(
+        [...new Set(qyDB.queryFiles("flights/**", { _from: "airports/LAX" }).map((x) => x.prop("_to"))).values()],
+        [
+            "airports/CVG",
+            "airports/SMF",
+            "airports/ATL",
+            "airports/MSP",
+            "airports/DFW",
+            "airports/PHL",
+            "airports/IAH",
+            "airports/TUS",
+            "airports/ORD",
+            "airports/PDX",
+            "airports/MIA",
+            "airports/SFO",
+            "airports/DEN",
+            "airports/IAD",
+            "airports/SJC",
+            "airports/PHX",
+            "airports/SEA",
+            "airports/SLC",
+            "airports/JFK",
+            "airports/SAN",
+            "airports/ONT",
+            "airports/LAS",
+            "airports/MCI",
+            "airports/MDW",
+            "airports/FLL",
+            "airports/AUS",
+            "airports/EWR",
+            "airports/CLE",
+            "airports/SBP",
+            "airports/MEM",
+            "airports/SBA",
+            "airports/CLT",
+            "airports/COS",
+            "airports/ASE",
+            "airports/ICT",
+            "airports/HOU",
+            "airports/YUM",
+            "airports/STL",
+            "airports/FAT",
+            "airports/ABQ",
+            "airports/BOI",
+            "airports/BOS",
+            "airports/LIH",
+            "airports/HNL",
+            "airports/SMX",
+            "airports/MRY",
+            "airports/DTW",
+            "airports/RNO",
+            "airports/MCO",
+            "airports/OXR",
+            "airports/OAK",
+            "airports/OGG",
+            "airports/KOA",
+            "airports/BWI",
+            "airports/CLD",
+            "airports/IND",
+            "airports/PSP",
+            "airports/ELP",
+            "airports/TPA",
+            "airports/OKC",
+            "airports/MSY",
+            "airports/SNA",
+            "airports/IYK",
+            "airports/IPL",
+            "airports/SAT",
+            "airports/MOD",
+            "airports/BNA",
+            "airports/DCA",
+            "airports/SGU",
+            "airports/BFL",
+            "airports/EUG",
+            "airports/EGE",
+            "airports/GEG",
+            "airports/BDL",
+            "airports/JAX",
+            "airports/CMH",
+            "airports/PIT",
+            "airports/RDU",
+            "airports/SJU",
+            "airports/XNA",
+            "airports/MTJ",
+            "airports/JAC",
+        ],
+    );
     console.timeEnd("query1");
 }
 
@@ -111,7 +110,7 @@ function query1(qyDB) {
 
 function query2(qyDB) {
     console.time("query2");
-    assert.deepEqual(
+    deepEqual(
         qyDB
             .queryFiles("flights", {
                 $and: [
@@ -120,19 +119,20 @@ function query2(qyDB) {
                     { Day: (x) => x >= 5 && x <= 7 },
                 ],
             })
-            .map((x) => x.fileContent.FlightNum),
+            .map((x) => x.prop("FlightNum"))
+            .sort((x, y) => x - y),
         [
             6635, 646, 6541, 5651, 6541, 5648, 687, 309, 6678, 6678, 6636, 439, 1567, 6635, 5647, 6541, 5660, 6541,
             6553, 687, 6553, 309, 5649, 5650, 6636, 439, 6635, 574, 6541, 5651, 5648, 6541, 1705, 1535, 5649, 6678,
             5650, 6636, 6678, 1087,
-        ],
+        ].sort((x, y) => x - y),
     );
     console.timeEnd("query2");
 }
 
 function query3(qyDB) {
     console.time("query3");
-    assert.deepEqual(
+    deepEqual(
         qyDB
             .queryFiles("flights", {
                 $or: [
@@ -142,7 +142,8 @@ function query3(qyDB) {
                 Month: 1,
                 Day: (x) => x >= 1 && x <= 7,
             })
-            .map((x) => x.view(["_from", "_to", "ArrTimeUTC"])),
+            .map((x) => x.view(["_from", "_to", "ArrTimeUTC"]))
+            .sort((x, y) => (x.ArrTimeUTC > y.ArrTimeUTC ? 1 : x.ArrTimeUTC == y.ArrTimeUTC ? 0 : -1)),
         [
             {
                 _from: "airports/BOI",
@@ -339,7 +340,7 @@ function query3(qyDB) {
                 _to: "airports/BOI",
                 ArrTimeUTC: "2008-01-08T06:29:00.000Z",
             },
-        ],
+        ].sort((x, y) => (x.ArrTimeUTC > y.ArrTimeUTC ? 1 : x.ArrTimeUTC == y.ArrTimeUTC ? 0 : -1)),
     );
 
     console.timeEnd("query3");
@@ -356,24 +357,27 @@ function query3(qyDB) {
 
 function query4(qyDB) {
     console.time("query4");
-    const result = qyDB
-        .queryFiles("flights", { Month: 1, Day: 5, FlightNum: 860 })
-        .map((x) => x.view(["_from", "_to"]));
-    assert.deepEqual(result, [
-        { _from: "airports/IAD", _to: "airports/BOS" },
-        { _from: "airports/PBI", _to: "airports/JFK" },
-        { _from: "airports/ORD", _to: "airports/DTW" },
-        { _from: "airports/PHX", _to: "airports/DEN" },
-        { _from: "airports/SEA", _to: "airports/ATL" },
-    ]);
+    deepEqual(
+        qyDB.queryFiles("flights", { Month: 1, Day: 5, FlightNum: 860 }).map((x) => x.view(["_from", "_to"])),
+        [
+            { _from: "airports/IAD", _to: "airports/BOS" },
+            { _from: "airports/PBI", _to: "airports/JFK" },
+            { _from: "airports/ORD", _to: "airports/DTW" },
+            { _from: "airports/PHX", _to: "airports/DEN" },
+            { _from: "airports/SEA", _to: "airports/ATL" },
+        ],
+    );
     console.timeEnd("query4");
 }
 
 function query5(qyDB) {
     logger.info("Query begins");
     console.time("query5");
-    assert.deepEqual(
-        qyDB.queryFiles("airports", { vip: true }).map((x) => x.view(["name"]).name),
+    deepEqual(
+        qyDB
+            .queryFiles("airports", { vip: true })
+            .map((x) => x.prop("name"))
+            .sort(),
         [
             "Amarillo International",
             "William B Hartsfield-Atlanta Intl",
@@ -382,7 +386,7 @@ function query5(qyDB) {
             "Los Angeles International",
             "Chicago O'Hare International",
             "San Francisco International",
-        ],
+        ].sort(),
     );
     console.timeEnd("query5");
 }
@@ -400,4 +404,4 @@ function runQueries(qyDB) {
     query5(qyDB);
 }
 
-Object.assign(module.exports, { runQueries });
+export { runQueries };