Add columns which start with 'Sitelink' to the new routes sheet (#89)

Co-authored-by: Kumaran Rajendhiran <[email protected]>
airtai · Aug 28, 2024 · 77a23f1 · 77a23f1
1 parent dcfcab0
commit 77a23f1
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/google_sheets/data_processing/processing.py b/google_sheets/data_processing/processing.py
@@ -71,6 +71,7 @@ def _validate_language_codes(
     "Include Location",
     "Include Language",
     "Exclude Language",
+    "Sitelink",
 ]
 
 

diff --git a/tests/data_processing/test_processing.py b/tests/data_processing/test_processing.py
@@ -353,6 +353,8 @@ def test_process_data_f(
                     "Include Location 2": "Croatia",
                     "Include Language 1": "English",
                     "Exclude Language 1": "German",
+                    "Sitelink Asset ID 1": "111",
+                    "Sitelink Asset ID 2": "222",
                 }
             ),
             pd.Series(
@@ -370,6 +372,8 @@ def test_process_data_f(
                     "Include Location 2": "Croatia",
                     "Include Language 1": "English",
                     "Exclude Language 1": "German",
+                    "Sitelink Asset ID 1": "111",
+                    "Sitelink Asset ID 2": "222",
                 }
             ),
         ),