scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-06 02:48:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

style: format tests/guardrails_tests/test_ztds_guardrail.py with ruff to pass CI lint gate

Browse source
This commit is contained in:
Ilya Sibiryakov 2026-09-27 09:53:22 +03:00
parent d85dbc773c
commit 75ceee8391
1 changed files with 7 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
tests/guardrails_tests/test_ztds_guardrail.py
View file

@ -5,6 +5,7 @@ https://datatracker.ietf.org/doc/draft-sibiryakov-ztds-protocol/
"""
import unittest
from litellm.proxy.guardrails.guardrail_hooks.ztds import ZTDSGuardrail
@ -58,7 +59,7 @@ class TestZTDSLiteLLMGuardrail(unittest.IsolatedAsyncioTestCase):
session_id = "test-session-1"
secret = "sk-" + "live12345678901234567890"
text = f"Contact alice@example.com or write to alice@example.com for secret {secret}."
sanitized, token_map = self.guardrail.sanitize_text(text, session_id)
sanitized, _ = self.guardrail.sanitize_text(text, session_id)
self.assertNotIn("alice@example.com", sanitized)
self.assertNotIn(secret, sanitized)
@ -76,7 +77,7 @@ class TestZTDSLiteLLMGuardrail(unittest.IsolatedAsyncioTestCase):
"""Invariant 1: Zero cleartext egress for emails, cards, phones, and API secrets."""
session_id = "test-session-2"
text = "Card 4111-2222-3333-4444 call +1-555-019-2834 server 192.168.1.100"
sanitized, token_map = self.guardrail.sanitize_text(text, session_id)
sanitized, _ = self.guardrail.sanitize_text(text, session_id)
self.assertIn("[CREDIT_CARD_TOKEN_1]", sanitized)
self.assertIn("[PHONE_TOKEN_1]", sanitized)
@ -88,7 +89,10 @@ class TestZTDSLiteLLMGuardrail(unittest.IsolatedAsyncioTestCase):
request_data = {
"litellm_call_id": "call-101",
"messages": [
{"role": "user", "content": "Please verify user bob@enterprise.corp with IBAN DE89370400440532013000"}
{
"role": "user",
"content": "Please verify user bob@enterprise.corp with IBAN DE89370400440532013000",
}
],
}