Extend lint test to test/ directory (#13247)

test/srt/openai_server/basic/test_openai_embedding.py

@@ -1,8 +1,6 @@
 import json
-import os
 import unittest
 
-import numpy as np
 import openai
 
 from sglang.srt.utils import kill_process_tree

test/srt/openai_server/basic/test_openai_server.py

@@ -7,12 +7,10 @@ python3 -m unittest openai_server.basic.test_openai_server.TestOpenAIServer.test
 
 import json
 import random
-import re
 import unittest
 from concurrent.futures import ThreadPoolExecutor
 from typing import Optional
 
-import numpy as np
 import openai
 import requests
 

test/srt/openai_server/basic/test_protocol.py

@@ -13,53 +13,19 @@
 # ==============================================================================
 """Tests for OpenAI API protocol models"""
 
-import json
-import time
 import unittest
-from typing import Dict, List, Optional
+from typing import List, Optional
 
 from pydantic import BaseModel, Field, ValidationError
 
 from sglang.srt.entrypoints.openai.protocol import (
-    BatchRequest,
-    BatchResponse,
-    ChatCompletionMessageContentImagePart,
-    ChatCompletionMessageContentTextPart,
     ChatCompletionRequest,
     ChatCompletionResponse,
     ChatCompletionResponseChoice,
-    ChatCompletionResponseStreamChoice,
-    ChatCompletionStreamResponse,
-    ChatCompletionTokenLogprob,
     ChatMessage,
-    ChoiceLogprobs,
     CompletionRequest,
-    CompletionResponse,
-    CompletionResponseChoice,
-    DeltaMessage,
-    EmbeddingObject,
-    EmbeddingRequest,
-    EmbeddingResponse,
-    ErrorResponse,
-    FileDeleteResponse,
-    FileRequest,
-    FileResponse,
-    Function,
-    FunctionResponse,
-    JsonSchemaResponseFormat,
-    LogProbs,
     ModelCard,
     ModelList,
-    MultimodalEmbeddingInput,
-    ResponseFormat,
-    ScoringRequest,
-    ScoringResponse,
-    StreamOptions,
-    StructuralTagResponseFormat,
-    Tool,
-    ToolCall,
-    ToolChoice,
-    TopLogprob,
     UsageInfo,
 )
 

test/srt/openai_server/basic/test_serving_completions.py

@@ -6,7 +6,7 @@ Run with:
 
 import unittest
 from typing import Optional
-from unittest.mock import AsyncMock, Mock, patch
+from unittest.mock import AsyncMock, Mock
 
 from sglang.srt.entrypoints.openai.protocol import CompletionRequest
 from sglang.srt.entrypoints.openai.serving_completions import OpenAIServingCompletion

test/srt/openai_server/basic/test_serving_embedding.py

@@ -10,7 +10,6 @@ from fastapi import Request
 
 from sglang.srt.entrypoints.openai.protocol import (
     EmbeddingRequest,
-    EmbeddingResponse,
     MultimodalEmbeddingInput,
 )
 from sglang.srt.entrypoints.openai.serving_embedding import OpenAIServingEmbedding

test/srt/openai_server/features/test_cache_report.py

@@ -1,4 +1,3 @@
-import asyncio
 import unittest
 
 import openai

test/srt/openai_server/features/test_enable_thinking.py

@@ -6,18 +6,12 @@ python3 -m unittest openai_server.features.test_enable_thinking.TestEnableThinki
 python3 -m unittest openai_server.features.test_enable_thinking.TestEnableThinking.test_stream_chat_completion_without_reasoning
 """
 
-import asyncio
 import json
-import os
-import sys
-import time
 import unittest
 
-import openai
 import requests
 
 from sglang.srt.utils import kill_process_tree
-from sglang.srt.utils.hf_transformers_utils import get_tokenizer
 from sglang.test.test_utils import (
     DEFAULT_ENABLE_THINKING_MODEL_NAME_FOR_TEST,
     DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,

test/srt/openai_server/features/test_openai_server_hidden_states.py

@@ -1,19 +1,13 @@
-import json
-import re
-import time
 import unittest
 from abc import ABC
 
-import numpy as np
 import openai
-import torch
 
 from sglang.srt.utils import kill_process_tree
 from sglang.srt.utils.hf_transformers_utils import get_tokenizer
 from sglang.test.test_utils import (
     DEFAULT_EAGLE_DRAFT_MODEL_FOR_TEST,
     DEFAULT_EAGLE_TARGET_MODEL_FOR_TEST,
-    DEFAULT_SMALL_EMBEDDING_MODEL_NAME_FOR_TEST,
     DEFAULT_SMALL_MODEL_NAME_FOR_TEST,
     DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
     DEFAULT_URL_FOR_TEST,

test/srt/openai_server/features/test_reasoning_content.py

@@ -9,11 +9,9 @@ python3 -m unittest openai_server.features.test_reasoning_content.TestReasoningC
 python3 -m unittest openai_server.features.test_reasoning_content.TestReasoningContentStartup.test_streaming
 """
 
-import json
 import unittest
 
 import openai
-import requests
 
 from sglang.srt.utils import kill_process_tree
 from sglang.test.test_utils import (

test/srt/openai_server/function_call/test_openai_function_calling.py

@@ -1,5 +1,4 @@
 import json
-import time
 import unittest
 
 import openai