Some test code might be written in just one line.

[dhr613]

current code:

  @patch("common.models.qwen.ChatQwQ")
  @patch.dict(os.environ, {"REGION": ""}, clear=False)
  def test_create_qwen_model_qwq(mock_chat_qwq):
      """Test QwQ model creation uses ChatQwQ."""
      mock_instance = Mock()
      mock_chat_qwq.return_value = mock_instance
  
      result = create_qwen_model("qwq-32b-preview", api_key="test-key")
  
      assert result == mock_instance
      mock_chat_qwq.assert_called_once_with(model="qwq-32b-preview", api_key="test-key")

simplified code

@patch("common.models.qwen.ChatQwQ")
@patch.dict(os.environ, {"REGION": ""}, clear=False)
def test_create_qwen_model_qwq(mock_chat_qwq):
    """Test QwQ model creation uses ChatQwQ."""
    assert mock_chat_qwq.return_value == create_qwen_model("qwq-32b-preview", api_key="test-key")
    mock_chat_qwq.assert_called_once_with(model="qwq-32b-preview", api_key="test-key")