Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import pytest
- from phone_parser import parse_phone_number
- def test_parse_phone_number():
- # Valid phone numbers
- assert parse_phone_number('555-123-4567') == {'area_code': '555', 'prefix': '123', 'line_number': '4567'}
- assert parse_phone_number('(555) 123-4567') == {'area_code': '555', 'prefix': '123', 'line_number': '4567'}
- assert parse_phone_number('555.123.4567') == {'area_code': '555', 'prefix': '123', 'line_number': '4567'}
- assert parse_phone_number('5551234567') == {'area_code': '555', 'prefix': '123', 'line_number': '4567'}
- assert parse_phone_number('1-555-123-4567') == {'area_code': '555', 'prefix': '123', 'line_number': '4567'}
- # Invalid phone numbers
- assert parse_phone_number('5551234') == None
- assert parse_phone_number('55512345678') == None
- assert parse_phone_number('1-555-1234-567') == None
- assert parse_phone_number('(555)123-4567') == None
- assert parse_phone_number('555-123-4567-') == None
Add Comment
Please, Sign In to add comment