Added test for JSON

Author: jaredoconnell

src/guidellm/__main__.py

@@ -289,7 +289,7 @@ def benchmark(
     default=Path.cwd() / "benchmarks.json",
 )
 def display(path):
-    asyncio.run(display_benchmarks_report(path))
+    display_benchmarks_report(path)
 
 
 def decode_escaped_str(_ctx, _param, value):

src/guidellm/benchmark/entrypoints.py

@@ -136,7 +136,7 @@ async def benchmark_generative_text(
 
     return report, saved_path
 
-async def display_benchmarks_report(file: str):
+def display_benchmarks_report(file: str):
     console = GenerativeBenchmarksConsole(enabled=True)
     if not os.path.exists(file):
         console.print_line(f"File {file} not found.")

tests/unit/entrypoints/__init__.py

@@ -0,0 +1,50 @@
+                    
+                    
+Benchmarks Metadata:
+    Run id:93e36b31-b454-471d-ba62-6b2671585485
+    Duration:30.2 seconds
+    Profile:type=sweep, strategies=['synchronous', 'throughput', 'constant',    
+    'constant', 'constant', 'constant', 'constant', 'constant', 'constant',     
+    'constant'], max_concurrency=None                                           
+    Args:max_number=None, max_duration=30.0, warmup_number=None,                
+    warmup_duration=None, cooldown_number=None, cooldown_duration=None          
+    Worker:type_='generative_requests_worker' backend_type='openai_http'        
+    backend_target='example_target' backend_model='example_model'               
+    backend_info={'max_output_tokens': 16384, 'timeout': 300, 'http2': True,    
+    'authorization': False, 'organization': None, 'project': None,              
+    'text_completions_path': '/v1/completions', 'chat_completions_path':        
+    '/v1/chat/completions'}                                                     
+    Request Loader:type_='generative_request_loader'                            
+    data='prompt_tokens=256,output_tokens=128' data_args=None                   
+    processor='example_processor' processor_args=None                           
+    Extras:None
+                
+                
+Benchmarks Info:
+================================================================================
+===================================================================             
+Metadata                                    |||| Requests Made  ||| Prompt      
+Tok/Req ||| Output Tok/Req  ||| Prompt Tok Total||| Output Tok Total  ||        
+  Benchmark| Start Time| End Time| Duration (s)|  Comp|  Inc|  Err|  Comp|      
+Inc| Err|   Comp|  Inc|  Err|   Comp|  Inc|  Err|   Comp|   Inc|   Err          
+-----------|-----------|---------|-------------|------|-----|-----|------|------
+|----|-------|-----|-----|-------|-----|-----|-------|------|------             
+synchronous|   16:59:28| 16:59:58|         30.0|    46|    1|    0| 257.1|      
+256.0| 0.0|  128.0|  0.0|  0.0|  11827|  256|    0|   5888|     0|     0        
+================================================================================
+===================================================================             
+                 
+                 
+Benchmarks Stats:
+================================================================================
+===============================================================                 
+Metadata   | Request Stats         || Out Tok/sec| Tot Tok/sec| Req Latency     
+(sec)  ||| TTFT (ms)       ||| ITL (ms)       ||| TPOT (ms)      ||             
+  Benchmark| Per Second| Concurrency|        mean|        mean|  mean|  median| 
+p99| mean| median|  p99| mean| median| p99| mean| median| p99                   
+-----------|-----------|------------|------------|------------|------|--------|-
+-----|-----|-------|-----|-----|-------|----|-----|-------|----                 
+synchronous|       1.55|        1.00|       198.1|       992.7|  0.64|    0.64| 
+0.69| 16.8|   16.4| 21.3|  4.9|    4.9| 5.3|  4.9|    4.9| 5.2                  
+================================================================================
+===============================================================                 

tests/unit/entrypoints/assets/benchmarks_stripped_output.txt

@@ -0,0 +1,26 @@
+import os
+import unittest
+import pytest
+
+from guidellm.benchmark import display_benchmarks_report
+
+@pytest.fixture()
+def get_test_asset_dir():
+    def _() -> str:
+        return os.path.dirname(os.path.abspath(__file__)) + "/assets"
+
+    return _
+
+
+def test_display_entrypoint_json(capfd, get_test_asset_dir):
+    asset_dir = get_test_asset_dir()
+    display_benchmarks_report(asset_dir + "/benchmarks_stripped.json")
+    out, err = capfd.readouterr()
+    expected_output_path = asset_dir + "/benchmarks_stripped_output.txt"
+    with open(expected_output_path, 'r', encoding='utf_8') as file:
+        expected_output = file.read()
+    assert out == expected_output
+
+
+if __name__ == '__main__':
+    unittest.main()