| # -*- coding: utf-8 -*- | |
| import pytest | |
| from pytube import helpers | |
| from pytube.exceptions import RegexMatchError | |
| def test_regex_search_no_match(): | |
| with pytest.raises(RegexMatchError): | |
| helpers.regex_search('^a$', '', groups=True) | |
| def test_regex_search(): | |
| # TODO(nficano): should check isinstance | |
| assert helpers.regex_search('^a$', 'a') is not None | |
| def test_safe_filename(): | |
| """Unsafe characters get stripped from generated filename""" | |
| assert helpers.safe_filename('abc1245$$') == 'abc1245' | |
| assert helpers.safe_filename('abc##') == 'abc' | |