From 0869068e7c99564c27f30ae87edd49ef70b7bdd4 Mon Sep 17 00:00:00 2001
From: Marut Pandya <pandyamarut@gmail.com>
Date: Mon, 31 Mar 2025 15:32:06 -0700
Subject: [PATCH] Create tests.json

---
 .runpod/tests.json | 47 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 47 insertions(+)
 create mode 100644 .runpod/tests.json

diff --git a/.runpod/tests.json b/.runpod/tests.json
new file mode 100644
index 0000000..279c007
--- /dev/null
+++ b/.runpod/tests.json
@@ -0,0 +1,47 @@
+{
+  // Test cases to validate the tool's functionality.
+  "tests": [
+    {
+      "name": "basic_inference_test",
+      // Basic inference with a simple prompt
+      "input": {
+        "prompt": "Write a short poem about artificial intelligence.",
+      },
+      "timeout": 30000
+    }
+  ],
+  // Test environment configuration settings.
+  "config": {
+    "gpuTypeId": "NVIDIA GeForce RTX 4090",
+    "gpuCount": 1,
+    "env": [
+      {
+        "key": "MODEL_NAME",
+        "value": "facebook/opt-350m"
+      },
+      {
+        "key": "HF_TOKEN",
+        "value": "hf_dummy_token_for_testing_purposes_only"
+      },
+      {
+        "key": "MAX_MODEL_LEN",
+        "value": "8192"
+      },
+      {
+        "key": "GPU_MEMORY_UTILIZATION",
+        "value": "0.95"
+      }
+    ],
+    "allowedCudaVersions": [
+      "12.7", 
+      "12.6", 
+      "12.5", 
+      "12.4", 
+      "12.3", 
+      "12.2", 
+      "12.1", 
+      "12.0", 
+      "11.7"
+    ]
+  }
+}