From 4fe62f34ed15e5933d0705d97d45aaa8fa74b840 Mon Sep 17 00:00:00 2001
From: Kye <kye@apacmediasolutions.com>
Date: Mon, 30 Oct 2023 14:43:38 -0400
Subject: [PATCH] clean up

---
 tests/structs/flow.py | 17 +++++++++--------
 1 file changed, 9 insertions(+), 8 deletions(-)

diff --git a/tests/structs/flow.py b/tests/structs/flow.py
index b9f251dc..74720b55 100644
--- a/tests/structs/flow.py
+++ b/tests/structs/flow.py
@@ -1,13 +1,13 @@
-import pytest
 import json
-from unittest.mock import MagicMock
-
-from unittest.mock import patch
 import os
-from swarms.structs.flow import Flow, stop_when_repeats
-from swarms.models import OpenAIChat
+from unittest.mock import MagicMock, patch
+
+import pytest
 from dotenv import load_dotenv
 
+from swarms.models import OpenAIChat
+from swarms.structs.flow import Flow, stop_when_repeats
+
 load_dotenv()
 
 openai_api_key = os.getenv("OPENAI_API_KEY")
@@ -89,7 +89,8 @@ def test_env_variable_handling(monkeypatch):
 
 # Test initializing the flow with different stopping conditions
 def test_flow_with_custom_stopping_condition(mocked_llm):
-    stopping_condition = lambda x: "terminate" in x.lower()
+    def stopping_condition(x):
+        return "terminate" in x.lower()
     flow = Flow(llm=mocked_llm, max_loops=5, stopping_condition=stopping_condition)
     assert flow.stopping_condition("Please terminate now")
     assert not flow.stopping_condition("Continue the process")
@@ -165,7 +166,7 @@ def test_mocked_openai_chat(MockedOpenAIChat):
     llm = MockedOpenAIChat(openai_api_key=openai_api_key)
     llm.return_value = MagicMock()
     flow = Flow(llm=llm, max_loops=5)
-    response = flow.run("Mocked run")
+    flow.run("Mocked run")
     assert MockedOpenAIChat.called
 
 # Test retry attempts