Refactor Locust testing to single test type

locust/README.md

@@ -8,30 +8,14 @@ In your Python environment with model dependencies installed, install the Locust
 pip install -r requirements.txt
 ```
 
-## Test Types
+## Load Test
 
-### Static Test
+Run a test using a dataset of variable input prompts generated by [generate_prompts](./../utils/prompt_generation.py#250).
 
-Run a basic test with a single user and a predefined input prompt.
-
-- Prompt: "What is in Austin Texas?"
-  - Prompt Length: 7 tokens
-  - Output Length: 128 tokens
-
-Command to run the static test:
-
-```bash
-locust --config locust_static.conf
-```
-
-### Dynamic Test
-
-Run a dynamic test using a dataset of variable input prompts generated by [generate_prompts](./../utils/prompt_generation.py#250).
-
-Command to run the dynamic test:
+Command to run the load test:
 
 ```bash
-locust --config locust_dynamic.conf
+locust --config locust_config.conf
 ```
 
 ## Test Configuration

locust/locust_dynamic.conf → locust/locust_config.conf

@@ -4,4 +4,3 @@ host = http://localhost:7000
 users = 32
 spawn-rate = 1
 run-time = 3m
-tags = [dynamic]

locust/locustfile.py

@@ -7,7 +7,7 @@
 import jwt
 
 from data_reader import DataReader
-from locust import FastHttpUser, events, tag, task
+from locust import FastHttpUser, events, task
 
 # Constants for timeouts and API configuration
 NETWORK_TIMEOUT = 300.0
@@ -39,13 +39,9 @@ def get_authorization():
 
 # Event listener to load custom data before tests start
 @events.test_start.add_listener
-def load_custom_data(environment, **kwargs):
+def load_custom_data(**kwargs):
     global data_iter
-    if "dynamic" in environment.parsed_options.tags:
-        data_iter = DataReader()
-        print("Dynamic data loaded.")
-    else:
-        print("Dynamic data not loaded (no 'dynamic' tag).")
+    data_iter = DataReader()
 
 
 class ServeUser(FastHttpUser):
@@ -64,16 +60,8 @@ def post_request(self, prompt: str, max_tokens: int):
         response = self.client.post(API_ENDPOINT, json=json_data, headers=self.headers)
         return response
 
-    @tag("static")
-    @task
-    def basic_test(self):
-        """Test using a static prompt."""
-        prompt = "What is in Austin Texas?"
-        self.post_request(prompt, max_tokens=128)
-
-    @tag("dynamic")
     @task
     def dataset_test(self):
-        """Test using dynamic prompts from a data iterator."""
+        """Test using generated prompts from a data iterator."""
         prompt = next(data_iter)
         self.post_request(prompt, max_tokens=128)