From d47b8a1ad531d7c17ae8772e04b773e7def9a521 Mon Sep 17 00:00:00 2001
From: Jennifer Melot <jtmelot@gmail.com>
Date: Tue, 17 Dec 2024 10:42:36 -0500
Subject: [PATCH] Add new classifiers to tests

---
 web/tests/test_data/alphabet_input.json  | 154 +++++++++++++++++++++++
 web/tests/test_data/alphabet_output.json |  72 +++++++++++
 2 files changed, 226 insertions(+)

diff --git a/web/tests/test_data/alphabet_input.json b/web/tests/test_data/alphabet_input.json
index 0aa7261..37c67a3 100644
--- a/web/tests/test_data/alphabet_input.json
+++ b/web/tests/test_data/alphabet_input.json
@@ -150,6 +150,8 @@
   "cv_pubs": 68,
   "nlp_pubs": 1,
   "robotics_pubs": 31,
+  "ai_safety_pubs": 31,
+  "llm_pubs": 31,
   "ai_pubs_by_year": [
     {
       "year": 2014,
@@ -302,6 +304,82 @@
       "num_papers": 10
     }
   ],
+  "ai_safety_pubs_by_year": [
+    {
+      "year": 2014,
+      "num_papers": 1
+    },
+    {
+      "year": 2015,
+      "num_papers": 0
+    },
+    {
+      "year": 2016,
+      "num_papers": 0
+    },
+    {
+      "year": 2017,
+      "num_papers": 1
+    },
+    {
+      "year": 2018,
+      "num_papers": 4
+    },
+    {
+      "year": 2019,
+      "num_papers": 4
+    },
+    {
+      "year": 2020,
+      "num_papers": 6
+    },
+    {
+      "year": 2021,
+      "num_papers": 5
+    },
+    {
+      "year": 2022,
+      "num_papers": 10
+    }
+  ],
+  "llm_pubs_by_year": [
+    {
+      "year": 2014,
+      "num_papers": 1
+    },
+    {
+      "year": 2015,
+      "num_papers": 0
+    },
+    {
+      "year": 2016,
+      "num_papers": 0
+    },
+    {
+      "year": 2017,
+      "num_papers": 1
+    },
+    {
+      "year": 2018,
+      "num_papers": 4
+    },
+    {
+      "year": 2019,
+      "num_papers": 4
+    },
+    {
+      "year": 2020,
+      "num_papers": 6
+    },
+    {
+      "year": 2021,
+      "num_papers": 5
+    },
+    {
+      "year": 2022,
+      "num_papers": 10
+    }
+  ],
   "highly_cited_ai_pubs_by_year": [
     {
       "year": 2014,
@@ -571,6 +649,82 @@
       "num_papers": 39
     }
   ],
+  "ai_safety_citation_count_by_year": [
+    {
+      "year": 2015,
+      "num_papers": 1
+    },
+    {
+      "year": 2016,
+      "num_papers": 15
+    },
+    {
+      "year": 2017,
+      "num_papers": 45
+    },
+    {
+      "year": 2018,
+      "num_papers": 230
+    },
+    {
+      "year": 2019,
+      "num_papers": 541
+    },
+    {
+      "year": 2020,
+      "num_papers": 1357
+    },
+    {
+      "year": 2021,
+      "num_papers": 2039
+    },
+    {
+      "year": 2022,
+      "num_papers": 1820
+    },
+    {
+      "year": 2023,
+      "num_papers": 39
+    }
+  ],
+  "llm_citation_count_by_year": [
+    {
+      "year": 2015,
+      "num_papers": 1
+    },
+    {
+      "year": 2016,
+      "num_papers": 15
+    },
+    {
+      "year": 2017,
+      "num_papers": 45
+    },
+    {
+      "year": 2018,
+      "num_papers": 230
+    },
+    {
+      "year": 2019,
+      "num_papers": 541
+    },
+    {
+      "year": 2020,
+      "num_papers": 1357
+    },
+    {
+      "year": 2021,
+      "num_papers": 2039
+    },
+    {
+      "year": 2022,
+      "num_papers": 1820
+    },
+    {
+      "year": 2023,
+      "num_papers": 39
+    }
+  ],
   "fields": [
     {
       "field_name": "Segmentation",
diff --git a/web/tests/test_data/alphabet_output.json b/web/tests/test_data/alphabet_output.json
index 546570e..76b2178 100644
--- a/web/tests/test_data/alphabet_output.json
+++ b/web/tests/test_data/alphabet_output.json
@@ -266,6 +266,40 @@
       "total": 6087,
       "isTopResearch": false
     },
+    "ai_safety_citation_counts": {
+      "counts": [
+        0,
+        1,
+        15,
+        45,
+        230,
+        541,
+        1357,
+        2039,
+        1820,
+        39,
+        0
+      ],
+      "total": 6087,
+      "isTopResearch": false
+    },
+    "llm_citation_counts": {
+      "counts": [
+        0,
+        1,
+        15,
+        45,
+        230,
+        541,
+        1357,
+        2039,
+        1820,
+        39,
+        0
+      ],
+      "total": 6087,
+      "isTopResearch": false
+    },
     "cv_publications": {
       "counts": [
         1,
@@ -323,6 +357,44 @@
       "growth": 44.44,
       "citations_per_article": 196.3548387096774
     },
+    "ai_safety_publications": {
+      "counts": [
+        1,
+        0,
+        0,
+        1,
+        4,
+        4,
+        6,
+        5,
+        10,
+        0,
+        0
+      ],
+      "total": 31,
+      "isTopResearch": true,
+      "growth": 44.44,
+      "citations_per_article": 196.3548387096774
+    },
+    "llm_publications": {
+      "counts": [
+        1,
+        0,
+        0,
+        1,
+        4,
+        4,
+        6,
+        5,
+        10,
+        0,
+        0
+      ],
+      "total": 31,
+      "isTopResearch": true,
+      "growth": 44.44,
+      "citations_per_article": 196.3548387096774
+    },
     "ai_citations_per_article": {
       "counts": [
         0.0,