# -*- 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'