File size: 573 Bytes
e7964fa |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
# -*- 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'
|