[TESTS][Fix these tests]

Author: kyegomez

asb_research.py

@@ -15,4 +15,4 @@
     task="Build a swarm to write a research paper on the topic of AI"
 )
 
-print(json.dumps(result, indent=2))
+print(json.dumps(result, indent=2))

heavy_swarm.py …gent/heavy_swarm_examples/heavy_swarm.pyheavy_swarm.py renamed to examples/multi_agent/heavy_swarm_examples/heavy_swarm.py heavy_swarm.py renamed to examples/multi_agent/heavy_swarm_examples/heavy_swarm.py

@@ -1,10 +1,8 @@
 import uuid
 
 from swarms.telemetry.main import (
-    generate_unique_identifier,
     generate_user_id,
     get_machine_id,
-    get_system_info,
 )
 
 
@@ -24,33 +22,6 @@ def test_get_machine_id():
     assert all(char in "0123456789abcdef" for char in machine_id)
 
 
-def test_get_system_info():
-    # Get system information and ensure it's a dictionary with expected keys
-    system_info = get_system_info()
-    assert isinstance(system_info, dict)
-    expected_keys = [
-        "platform",
-        "platform_release",
-        "platform_version",
-        "architecture",
-        "hostname",
-        "ip_address",
-        "mac_address",
-        "processor",
-        "python_version",
-    ]
-    assert all(key in system_info for key in expected_keys)
-
-
-def test_generate_unique_identifier():
-    # Generate unique identifiers and ensure they are valid UUID strings
-    unique_id = generate_unique_identifier()
-    assert isinstance(unique_id, str)
-    assert uuid.UUID(
-        unique_id, version=5, namespace=uuid.NAMESPACE_DNS
-    )
-
-
 def test_generate_user_id_edge_case():
     # Test generate_user_id with multiple calls
     user_ids = set()
@@ -69,33 +40,13 @@ def test_get_machine_id_edge_case():
     assert len(machine_ids) == 100  # Ensure generated IDs are unique
 
 
-def test_get_system_info_edge_case():
-    # Test get_system_info for consistency
-    system_info1 = get_system_info()
-    system_info2 = get_system_info()
-    assert (
-        system_info1 == system_info2
-    )  # Ensure system info remains the same
-
-
-def test_generate_unique_identifier_edge_case():
-    # Test generate_unique_identifier for uniqueness
-    unique_ids = set()
-    for _ in range(100):
-        unique_id = generate_unique_identifier()
-        unique_ids.add(unique_id)
-    assert len(unique_ids) == 100  # Ensure generated IDs are unique
-
 
 def test_all():
     test_generate_user_id()
     test_get_machine_id()
-    test_get_system_info()
-    test_generate_unique_identifier()
     test_generate_user_id_edge_case()
     test_get_machine_id_edge_case()
-    test_get_system_info_edge_case()
-    test_generate_unique_identifier_edge_case()
+
 
 
 test_all()

tests/prompts/test_prompt.py

@@ -1,37 +1,29 @@
-import os
 import json
+import os
 from datetime import datetime
-from typing import List, Dict, Any, Callable
+from typing import Any, Callable, Dict, List
 
 from dotenv import load_dotenv
+from loguru import logger
 
 # Basic Imports for Swarms
 from swarms.structs import (
     Agent,
-    SequentialWorkflow,
-    ConcurrentWorkflow,
     AgentRearrange,
-    MixtureOfAgents,
-    SpreadSheetSwarm,
+    ConcurrentWorkflow,
     GroupChat,
-    MultiAgentRouter,
+    InteractiveGroupChat,
     MajorityVoting,
-    SwarmRouter,
+    MixtureOfAgents,
+    MultiAgentRouter,
     RoundRobinSwarm,
-    InteractiveGroupChat,
+    SequentialWorkflow,
+    SpreadSheetSwarm,
+    SwarmRouter,
 )
-
-# Import swarms not in __init__.py directly
 from swarms.structs.hiearchical_swarm import HierarchicalSwarm
 from swarms.structs.tree_swarm import ForestSwarm, Tree, TreeAgent
 
-# Setup Logging
-from loguru import logger
-
-logger.add(
-    "test_runs/test_failures.log", rotation="10 MB", level="ERROR"
-)
-
 # Load environment variables
 load_dotenv()
 
@@ -463,8 +455,8 @@ def test_spreadsheet_swarm():
 def test_hierarchical_swarm():
     """Test HierarchicalSwarm structure"""
     try:
-        from swarms.utils.litellm_wrapper import LiteLLM
         from swarms.structs.hiearchical_swarm import SwarmSpec
+        from swarms.utils.litellm_wrapper import LiteLLM
 
         # Create worker agents
         workers = [

tests/telemetry/test_user_utils.py

@@ -3,14 +3,6 @@
 
 load_dotenv()
 
-## [OPTIONAL] REGISTER MODEL - not all ollama models support function calling, litellm defaults to json mode tool calls if native tool calling not supported.
-
-# litellm.register_model(model_cost={
-#                 "ollama_chat/llama3.1": {
-#                   "supports_function_calling": true
-#                 },
-#             })
-
 tools = [
     {
         "type": "function",

tests/test_comprehensive_test.py

@@ -1,6 +1,3 @@
-#!/usr/bin/env python3
-"""Test script to verify the improved formatter markdown rendering."""
-
 from swarms.utils.formatter import Formatter
 
 

tests/utils/test_output_str_fix.py tests/tools/test_output_str_fix.pytests/utils/test_output_str_fix.py renamed to tests/tools/test_output_str_fix.py

@@ -1,21 +1,9 @@
 import asyncio
-import sys
 
 from loguru import logger
 
 from swarms.utils.litellm_wrapper import LiteLLM
 
-# Configure loguru logger
-logger.remove()  # Remove default handler
-logger.add(
-    "test_litellm.log",
-    rotation="1 MB",
-    format="{time} | {level} | {message}",
-    level="DEBUG",
-)
-logger.add(sys.stdout, level="INFO")
-
-
 tools = [
     {
         "type": "function",