File size: 1,612 Bytes
d6ea71e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 |
import os
from urllib.error import HTTPError
import pytest
from socceraction.data.base import (
_auth_remoteloadjson,
_has_auth,
_localloadjson,
_remoteloadjson,
)
def test_load_json_from_url() -> None:
url = "https://raw.githubusercontent.com/statsbomb/open-data/master/data/events/15946.json"
result = _remoteloadjson(url)
assert isinstance(result, list)
assert isinstance(result[0], dict)
def test_has_auth() -> None:
assert not _has_auth({})
assert not _has_auth({"user": "", "passwd": "test_passwd"})
assert not _has_auth({"user": "test_user"})
assert not _has_auth({"passwd": "test_passwd"})
assert _has_auth({"user": "test_user", "passwd": "test_passwd"})
def test_load_json_from_url_with_auth() -> None:
# use httpbin.org to test authentication
user = "test_user"
passwd = "test_passwd"
url = f"http://httpbin.org/basic-auth/{user}/{passwd}"
# add authentication header
_auth_remoteloadjson(user, passwd)
# the remote_load_json header should now use the authentication header
try:
result = _remoteloadjson(url)
assert isinstance(result, dict)
assert result.get("authenticated") is True
except HTTPError as e:
if e.code == 504:
pytest.skip("httpbin.org is down or too slow")
def test_load_json_from_file() -> None:
data_dir = os.path.join(os.path.dirname(__file__), os.pardir, "datasets", "statsbomb", "raw")
result = _localloadjson(os.path.join(data_dir, "events", "15946.json"))
assert isinstance(result, list)
assert isinstance(result[0], dict)
|