chore: bring accuracy tests from the kanopy backend (#4913)

* chore: bring accuracy tests from the kanopy backend * allow to pass a different backend url * Update scripts/ai-accuracy-tests/index.js Co-authored-by: Anna Henningsen <[email protected]> * Update scripts/ai-accuracy-tests/index.js Co-authored-by: Anna Henningsen <[email protected]> * Update scripts/ai-accuracy-tests/index.js Co-authored-by: Anna Henningsen <[email protected]> --------- Co-authored-by: Anna Henningsen <[email protected]>
mongodb-js · Oct 3, 2023 · 4d0e659 · 4d0e659
1 parent 1e2dcf3
commit 4d0e659
Show file tree

Hide file tree

Showing 8 changed files with 6,024 additions and 286 deletions.
diff --git a/package-lock.json b/package-lock.json
diff --git a/scripts/ai-accuracy-tests/fixtures/NYC.parking_2015.json b/scripts/ai-accuracy-tests/fixtures/NYC.parking_2015.json
diff --git a/scripts/ai-accuracy-tests/fixtures/berlin.cocktailbars.json b/scripts/ai-accuracy-tests/fixtures/berlin.cocktailbars.json
@@ -0,0 +1,113 @@
+{
+  "indexes": [{ "koordinaten": "2dsphere" }],
+  "data": [
+    {
+      "_id": {
+        "$oid": "5ca652bf56618187558b4de3"
+      },
+      "name": "Bar Zentral",
+      "strasse": "Lotte-Lenya-Bogen",
+      "hausnummer": 551,
+      "plz": 10623,
+      "webseite": "barzentralde",
+      "koordinaten": [13.404954, 52.520008]
+    },
+    {
+      "_id": {
+        "$oid": "5ca6544a97aed3878f9b090f"
+      },
+      "name": "Hefner Bar",
+      "strasse": "Kantstr.",
+      "hausnummer": 146,
+      "plz": 10623,
+      "koordinaten": [13.3213093, 42.5055506]
+    },
+    {
+      "_id": {
+        "$oid": "5ca654ec97aed3878f9b0910"
+      },
+      "name": "Bar am Steinplatz",
+      "strasse": "Steinplatz",
+      "hausnummer": 4,
+      "plz": 10623,
+      "webseite": "barsteinplatz.com",
+      "koordinaten": [13.3241804, 42.5081672]
+    },
+    {
+      "_id": {
+        "$oid": "5ca6559e97aed3878f9b0911"
+      },
+      "name": "Rum Trader",
+      "strasse": "Fasanenstr.",
+      "hausnummer": 40,
+      "plz": 10719,
+      "koordinaten": [13.3244667, 42.4984012]
+    },
+    {
+      "_id": {
+        "$oid": "5ca655f597aed3878f9b0912"
+      },
+      "name": "Stairs",
+      "strasse": "Uhlandstr.",
+      "hausnummer": 133,
+      "plz": 10717,
+      "webseite": "stairsbar-berlin.com",
+      "koordinaten": [13.3215159, 42.49256]
+    },
+    {
+      "_id": {
+        "$oid": "5ca656a697aed3878f9b0913"
+      },
+      "name": "Green Door",
+      "strasse": "Winterfeldtstr.",
+      "hausnummer": 50,
+      "plz": 10781,
+      "webseite": "greendoor.de",
+      "koordinaten": [13.3507105, 42.4970952]
+    },
+    {
+      "_id": {
+        "$oid": "5ca6570597aed3878f9b0914"
+      },
+      "name": "Mister Hu",
+      "strasse": "Goltzstr.",
+      "hausnummer": 39,
+      "plz": 10781,
+      "webseite": "misterhu.de",
+      "koordinaten": [13.3511185, 42.4927243]
+    },
+    {
+      "_id": {
+        "$oid": "5ca6576f97aed3878f9b0915"
+      },
+      "name": "Salut!",
+      "strasse": "Goltzstr.",
+      "hausnummer": 7,
+      "plz": 10781,
+      "webseite": "salut-berlin.de",
+      "koordinaten": [13.3513021, 42.4911044]
+    },
+    {
+      "_id": {
+        "$oid": "5ca6581197aed3878f9b0916"
+      },
+      "name": "Lebensstern",
+      "strasse": "Kurfürstenstr.",
+      "hausnummer": 58,
+      "plz": 10785,
+      "webseite": "lebens-stern.de",
+      "koordinaten": [13.3524999, 42.502059]
+    },
+    {
+      "_id": {
+        "$oid": "5ca6588397aed3878f9b0917"
+      },
+      "name": "Victoria Bar",
+      "strasse": "Potsdamer Str.",
+      "hausnummer": 102,
+      "plz": 10785,
+      "webseite": "victoriabar.de",
+      "koordinaten": [13.3616635, 42.5014176]
+    }
+  ]
+}
diff --git a/scripts/ai-accuracy-tests/fixtures/delimiters.filter.json b/scripts/ai-accuracy-tests/fixtures/delimiters.filter.json
@@ -0,0 +1,12 @@
+{
+  "data": [
+    {
+      "_id": "1",
+      "filter": true
+    },
+    {
+      "_id": "2",
+      "filter": false
+    }
+  ]
+}
diff --git a/scripts/ai-accuracy-tests/fixtures/netflix.movies.json b/scripts/ai-accuracy-tests/fixtures/netflix.movies.json
@@ -0,0 +1,124 @@
+{
+  "data": [
+    {
+      "_id": {
+        "$oid": "573b864df29313caabe354fc"
+      },
+      "title": "Nature: Antarctica",
+      "year": "1982",
+      "id": "14"
+    },
+    {
+      "_id": {
+        "$oid": "573b864df29313caabe3550d"
+      },
+      "title": "Ferngully 2: The Magical Rescue",
+      "year": "2000",
+      "id": "35"
+    },
+    {
+      "_id": {
+        "$oid": "573b864df29313caabe35513"
+      },
+      "title": "Love Reinvented",
+      "year": "2000",
+      "id": "39"
+    },
+    {
+      "_id": {
+        "$oid": "573b864df29313caabe3555b"
+      },
+      "title": "The Eye of Vichy",
+      "year": "1993",
+      "id": "112"
+    },
+    {
+      "_id": {
+        "$oid": "573b864df29313caabe3557a"
+      },
+      "title": "The Tricky Master",
+      "year": "2000",
+      "id": "142"
+    },
+    {
+      "_id": {
+        "$oid": "573b864df29313caabe35581"
+      },
+      "title": "A Little Princess",
+      "year": "1995",
+      "id": "152"
+    },
+    {
+      "_id": {
+        "$oid": "573b864df29313caabe35593"
+      },
+      "title": "Smokey and the Bandit Part 3",
+      "year": "1983",
+      "id": "168"
+    },
+    {
+      "_id": {
+        "$oid": "573b864df29313caabe355a8"
+      },
+      "title": "Airplane II: The Sequel",
+      "year": "1982",
+      "id": "189"
+    },
+    {
+      "_id": {
+        "$oid": "573b864df29313caabe355f1"
+      },
+      "title": "The Big Clock",
+      "year": "1948",
+      "id": "261"
+    },
+    {
+      "_id": {
+        "$oid": "573b864df29313caabe355fc"
+      },
+      "title": "Female Yakuza Tale",
+      "year": "1973",
+      "id": "272"
+    },
+    {
+      "_id": {
+        "$oid": "573b864df29313caabe3695c"
+      },
+      "title": "Alien 3000",
+      "year": "2004",
+      "id": "5233"
+    },
+    {
+      "_id": {
+        "$oid": "573b864ef29313caabe3907a"
+      },
+      "title": "Alien 3: Collector's Edition",
+      "year": "1992",
+      "id": "15246"
+    },
+    {
+      "_id": {
+        "$oid": "573b864df29313caabe37495"
+      },
+      "title": "Alien Apocalypse",
+      "year": "2005",
+      "id": "8106"
+    },
+    {
+      "_id": {
+        "$oid": "573b864ef29313caabe39507"
+      },
+      "title": "Alien Chaser",
+      "year": "1996",
+      "id": "16413"
+    },
+    {
+      "_id": {
+        "$oid": "573b864df29313caabe36f05"
+      },
+      "title": "Alien Files",
+      "year": "1999",
+      "id": "6682"
+    }
+  ]
+}