| import pytest | |
| from pytube import cipher | |
| from pytube.exceptions import RegexMatchError | |
| def test_map_functions(): | |
| with pytest.raises(RegexMatchError): | |
| cipher.map_functions("asdf") | |
| def test_get_initial_function_name_with_no_match_should_error(): | |
| with pytest.raises(RegexMatchError): | |
| cipher.get_initial_function_name("asdf") | |
| def test_get_transform_object_with_no_match_should_error(): | |
| with pytest.raises(RegexMatchError): | |
| cipher.get_transform_object("asdf", var="lt") | |
| def test_reverse(): | |
| reversed_array = cipher.reverse([1, 2, 3, 4], None) | |
| assert reversed_array == [4, 3, 2, 1] | |
| def test_splice(): | |
| assert cipher.splice([1, 2, 3, 4], 2) == [3, 4] | |
| assert cipher.splice([1, 2, 3, 4], 1) == [2, 3, 4] | |
| def test_throttling_reverse(): | |
| a = [1, 2, 3, 4] | |
| cipher.throttling_reverse(a) | |
| assert a == [4, 3, 2, 1] | |
| def test_throttling_push(): | |
| a = [1, 2, 3, 4] | |
| cipher.throttling_push(a, 5) | |
| assert a == [1, 2, 3, 4, 5] | |
| def test_throttling_unshift(): | |
| a = [1, 2, 3, 4] | |
| cipher.throttling_unshift(a, 2) | |
| assert a == [3, 4, 1, 2] | |
| def test_throttling_nested_splice(): | |
| a = [1, 2, 3, 4] | |
| cipher.throttling_nested_splice(a, 2) | |
| assert a == [3, 2, 1, 4] | |
| cipher.throttling_nested_splice(a, 0) | |
| assert a == [3, 2, 1, 4] | |
| def test_throttling_prepend(): | |
| a = [1, 2, 3, 4] | |
| cipher.throttling_prepend(a, 1) | |
| assert a == [4, 1, 2, 3] | |
| a = [1, 2, 3, 4] | |
| cipher.throttling_prepend(a, 2) | |
| assert a == [3, 4, 1, 2] | |
| def test_throttling_swap(): | |
| a = [1, 2, 3, 4] | |
| cipher.throttling_swap(a, 3) | |
| assert a == [4, 2, 3, 1] | |
| def test_js_splice(): | |
| mapping = { | |
| } | |
| for args, result in mapping.items(): | |
| a = [1, 2, 3, 4] | |
| assert cipher.js_splice(a, *args) == result | |