Checkpoint - working on numpy test variants

Author: caseyclements

.evergreen/generated_configs/tasks.yml

@@ -4768,6 +4768,14 @@ tasks:
       - noauth
       - pypy
 
+  # Test numpy tests
+  - name: test-numpy
+    commands:
+      - func: run test numpy
+        vars:
+          TEST_NAME: test_numpy
+    tags: [binary, vector]
+
   # Test standard auth tests
   - name: test-standard-auth-v4.2-python3.10-auth-ssl-sharded-cluster
     commands:

.evergreen/generated_configs/variants.yml

@@ -612,3 +612,46 @@ buildvariants:
       - rhel87-small
     expansions:
       STORAGE_ENGINE: inmemory
+
+  # Test numpy tests
+  - name: test-numpy-rhel8
+    tasks:
+      - name: .test-numpy
+    display_name: Test Numpy RHEL8
+    run_on:
+      - rhel87-small
+    tags: [binary-vector]
+  - name: test-numpy-macos
+    tasks:
+      - name: .test-numpy
+    display_name: Test Numpy macOS
+    run_on:
+      - macos-14
+    tags: [binary-vector]
+  - name: test-numpy-macos-arm64
+    tasks:
+      - name: .test-numpy !.pypy .server-6.0
+      - name: .test-numpy !.pypy .server-7.0
+      - name: .test-numpy !.pypy .server-8.0
+      - name: .test-numpy !.pypy .server-rapid
+      - name: .test-numpy !.pypy .server-latest
+    display_name: Test Numpy macOS Arm64
+    run_on:
+      - macos-14-arm64
+    tags: [binary-vector]
+  - name: test-numpy-win64
+    tasks:
+      - name: .test-numpy
+    display_name: Test Numpy Win64
+    run_on:
+      - windows-64-vsMulti-small
+    tags: [binary-vector]
+  - name: test-numpy-win32
+    tasks:
+      - name: .test-numpy
+    display_name: Test Numpy Win32
+    run_on:
+      - windows-64-vsMulti-small
+    expansions:
+      IS_WIN32: "1"
+    tags: [binary-vector]

.evergreen/scripts/generate_config.py

@@ -323,6 +323,38 @@ def create_disable_test_commands_variants():
     return [create_variant(tasks, display_name, host=host, expansions=expansions)]
 
 
+def create_test_numpy_tasks():
+    vars = dict(TEST_NAME="test_numpy")
+    test_func = FunctionCall(func="run test numpy", vars=vars)
+    task_name = "test-numpy"
+    tags = ["binary", "vector"]
+    return [EvgTask(name=task_name, tags=tags, commands=[test_func])]
+
+
+def create_test_numpy_variants() -> list[BuildVariant]:
+    variants = []
+    base_display_name = "Test Numpy"
+
+    # Test a subset on each of the other platforms.
+    for host_name in ("rhel8", "macos", "macos-arm64", "win64", "win32"):
+        tasks = [".test-numpy"]
+        # MacOS arm64 only works on server versions 6.0+
+        if host_name == "macos-arm64":
+            tasks = [
+                f".test-numpy !.pypy .server-{version}" for version in get_versions_from("6.0")
+            ]
+        host = HOSTS[host_name]
+        tags = ["binary-vector"]
+        expansions = dict()
+        if host_name == "win32":
+            expansions["IS_WIN32"] = "1"
+        display_name = get_variant_name(base_display_name, host)
+        variant = create_variant(tasks, display_name, host=host, tags=tags, expansions=expansions)
+        variants.append(variant)
+
+    return variants
+
+
 def create_oidc_auth_variants():
     variants = []
     for host_name in ["ubuntu22", "macos", "win64"]:
@@ -1174,3 +1206,6 @@ def create_send_dashboard_data_func():
 write_variants_to_file(mod)
 write_tasks_to_file(mod)
 write_functions_to_file(mod)
+
+# TODO - Create a new variant here that drives run-test
+#   Workfromrove

.evergreen/scripts/run_tests.py

@@ -214,3 +214,7 @@ def run() -> None:
 
 if __name__ == "__main__":
     run()
+
+
+# TODO - Make changes here to create a pytest marker that runs as desired
+#   Use encryption as example