Paul Bailey
77437348ed
- Created RoleDetector class with colon and bracket pattern extraction - Implemented role deduplication logic - Added fallback to default 'Narrator' role when no patterns found - Created /api/tts/roles endpoint using RoleDetector - Added comprehensive test suite for role detection - All tests passing (7/7) Requirements: 2.1, 2.2, 2.3, 2.4, 2.5, 2.6
106 lines
2.8 KiB
Python
106 lines
2.8 KiB
Python
#!/usr/bin/env python3
|
|
"""Test script for role detection service."""
|
|
import urllib.request
|
|
import json
|
|
|
|
|
|
def test_role_detection(text, expected_roles, description):
|
|
"""Test role detection with given text."""
|
|
print(f"\nTesting: {description}")
|
|
print(f"Input text: {text[:100]}..." if len(text) > 100 else f"Input text: {text}")
|
|
|
|
data = json.dumps({"text": text}).encode('utf-8')
|
|
req = urllib.request.Request(
|
|
'http://localhost:8001/api/tts/roles',
|
|
data=data,
|
|
headers={'Content-Type': 'application/json'}
|
|
)
|
|
|
|
try:
|
|
response = urllib.request.urlopen(req)
|
|
result = json.loads(response.read().decode())
|
|
roles = [role['label'] for role in result['roles']]
|
|
|
|
print(f"Detected roles: {roles}")
|
|
print(f"Expected roles: {expected_roles}")
|
|
|
|
if set(roles) == set(expected_roles):
|
|
print("✓ PASS")
|
|
return True
|
|
else:
|
|
print("✗ FAIL: Roles don't match")
|
|
return False
|
|
except Exception as e:
|
|
print(f"✗ FAIL: {str(e)}")
|
|
return False
|
|
|
|
|
|
def main():
|
|
"""Run all role detection tests."""
|
|
print("Testing Role Detection Service...")
|
|
|
|
tests = [
|
|
# Test colon pattern
|
|
(
|
|
"Alice: Hello there!\nBob: Hi Alice!",
|
|
["Alice", "Bob"],
|
|
"Colon pattern extraction"
|
|
),
|
|
# Test bracket pattern
|
|
(
|
|
"[Narrator] The story begins.\n[Alice] She said hello.",
|
|
["Alice", "Narrator"],
|
|
"Bracket pattern extraction"
|
|
),
|
|
# Test mixed patterns
|
|
(
|
|
"Alice: Hello!\n[Narrator] She waved.\nBob: Hey!",
|
|
["Alice", "Bob", "Narrator"],
|
|
"Mixed colon and bracket patterns"
|
|
),
|
|
# Test deduplication
|
|
(
|
|
"Alice: First line.\nAlice: Second line.\nAlice: Third line.",
|
|
["Alice"],
|
|
"Role deduplication"
|
|
),
|
|
# Test empty text
|
|
(
|
|
"",
|
|
["Narrator"],
|
|
"Empty text returns default Narrator"
|
|
),
|
|
# Test no role markers
|
|
(
|
|
"This is just plain text without any role markers.",
|
|
["Narrator"],
|
|
"No role markers returns default Narrator"
|
|
),
|
|
# Test whitespace only
|
|
(
|
|
" \n\n ",
|
|
["Narrator"],
|
|
"Whitespace only returns default Narrator"
|
|
),
|
|
]
|
|
|
|
passed = 0
|
|
failed = 0
|
|
|
|
for text, expected, description in tests:
|
|
if test_role_detection(text, expected, description):
|
|
passed += 1
|
|
else:
|
|
failed += 1
|
|
|
|
print(f"\n{'='*50}")
|
|
print(f"Results: {passed} passed, {failed} failed")
|
|
print(f"{'='*50}")
|
|
|
|
return failed == 0
|
|
|
|
|
|
if __name__ == '__main__':
|
|
success = main()
|
|
exit(0 if success else 1)
|