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