scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-08 22:21:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 78

fix: Correct test assertion bug and inaccurate docstring

Browse source 
Fix test_utorid_masked checking `result` (dict) instead of `output` (string).
Update test_ca_policy_e2e.py docstring to clarify scope vs UofT tests.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
Krrish Dholakia 2026-03-05 20:02:05 -08:00
parent d35a95fc5d
commit 74ed48d5d7
2 changed files with 5 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
tests/test_litellm/proxy/guardrails/guardrail_hooks/content_filter/test_ca_policy_e2e.py
View file

@ -1,9 +1,10 @@
"""
End-to-end tests for Canadian PII Protection (PIPEDA) policy template.
Tests the complete policy with all Canadian PII patterns — validates that
PII-containing prompts are detected/masked and that clean prompts pass through.
These tests mirror the canadianPiiCompliancePrompts.ts dataset.
Tests the federal/provincial PII patterns (SIN, OHIP, driver's licence, passport,
immigration docs, bank account, postal code) — validates that PII-containing prompts
are detected/masked and that clean prompts pass through.
University of Toronto institutional identifiers are tested separately in test_uoft_policy_e2e.py.
"""
import pytest

2
tests/test_litellm/proxy/guardrails/guardrail_hooks/content_filter/test_uoft_policy_e2e.py
View file

@ -114,7 +114,7 @@ class TestUofTPolicyE2E:
)
output = result.get("texts", [])[0]
assert "[UOFT_UTORID_REDACTED]" in output
assert "smithj12" not in result
assert "smithj12" not in output
@pytest.mark.asyncio
async def test_utorid_quercus_context_masked(self):