rename util tests

Author: Andrei Bratu

tests/utilities/test_flow_decorator.py tests/utilities/test_flow.pytests/utilities/test_flow_decorator.py renamed to tests/utilities/test_flow.py

@@ -65,7 +65,7 @@ def _flow_over_flow(messages: list[dict]) -> str:
     return _random_string, _call_llm, _agent_call, _flow_over_flow
 
 
-@pytest.mark.skip(reason="Runs single or as part of the suite, fails on a full run. Likely test config issue.")
+@pytest.mark.flaky(retries=3, delay=5)
 def test_decorators_without_flow(
     opentelemetry_hl_test_configuration: tuple[Tracer, InMemorySpanExporter],
 ):
@@ -89,7 +89,7 @@ def test_decorators_without_flow(
     # WHEN exporting the spans
     # Wait for the prompt span to be exported; It was waiting
     # on the OpenAI call span to finish first
-    time.sleep(1)
+    time.sleep(3)
     spans = exporter.get_finished_spans()
 
     # THEN 3 spans arrive at the exporter
@@ -111,6 +111,7 @@ def test_decorators_without_flow(
     )["prompt"]
 
 
+@pytest.mark.flaky(retries=3, delay=5)
 def test_decorators_with_flow_decorator(
     opentelemetry_hl_test_configuration: tuple[Tracer, InMemorySpanExporter],
 ):
@@ -133,6 +134,8 @@ def test_decorators_with_flow_decorator(
         ]
     )
 
+    time.sleep(3)
+
     # THEN 4 spans arrive at the exporter
     spans = exporter.get_finished_spans()
     assert len(spans) == 4