Fix and enable CSV output test

Signed-off-by: Jared O'Connell <[email protected]>

Author: jaredoconnell

tests/unit/benchmark/test_output.py

@@ -80,7 +80,6 @@ def test_file_yaml():
 
     mock_path.unlink()
 
-@pytest.mark.skip(reason="CSV fix not merged yet")
 @pytest.mark.asyncio
 async def test_file_csv():
     mock_benchmark = mock_generative_benchmark()
@@ -96,6 +95,7 @@ async def test_file_csv():
         rows = list(reader)
 
     assert "Type" in headers
+    assert "Profile" in headers
     assert len(rows) == 1
 
     mock_path.unlink()

tests/unit/mock_benchmark.py

@@ -76,7 +76,11 @@ def mock_generative_benchmark() -> GenerativeBenchmark:
         ),
         benchmarker=BenchmarkerDict(
             profile=SynchronousProfile.create("synchronous", rate=None),
-            requests={},
+            requests={
+                "attributes": {
+                    "data": "prompt_tokens=256,output_tokens=128",
+                },
+            },
             backend={},
             environment={},
             aggregators={},