pyodide/packages/micropip/test_micropip.py

107 lines
3.3 KiB
Python

import time
import sys
from pathlib import Path
import pytest
sys.path.append(str(Path(__file__).resolve().parent / "micropip"))
def test_install_simple(selenium_standalone):
selenium_standalone.run("import os")
selenium_standalone.load_package("micropip")
selenium_standalone.run("import micropip")
selenium_standalone.run("micropip.install('pyodide-micropip-test')")
# Package 'pyodide-micropip-test' has dependency on 'snowballstemmer'
# It is used to test markers support
for i in range(10):
if selenium_standalone.run(
"os.path.exists" "('/lib/python3.8/site-packages/snowballstemmer')"
):
break
else:
time.sleep(1)
selenium_standalone.run("import snowballstemmer")
selenium_standalone.run("stemmer = snowballstemmer.stemmer('english')")
assert selenium_standalone.run(
"stemmer.stemWords('go going goes gone'.split())"
) == ["go", "go", "goe", "gone"]
def test_parse_wheel_url():
pytest.importorskip("distlib")
import micropip
url = "https://a/snowballstemmer-2.0.0-py2.py3-none-any.whl"
name, wheel, version = micropip._parse_wheel_url(url)
assert name == "snowballstemmer"
assert version == "2.0.0"
assert wheel == {
"digests": None,
"filename": "snowballstemmer-2.0.0-py2.py3-none-any.whl",
"packagetype": "bdist_wheel",
"python_version": "py2.py3",
"abi_tag": "none",
"platform": "any",
"url": url,
}
msg = "not a valid wheel file name"
with pytest.raises(ValueError, match=msg):
url = "https://a/snowballstemmer-2.0.0-py2.whl"
name, params, version = micropip._parse_wheel_url(url)
url = "http://scikit_learn-0.22.2.post1-cp35-cp35m-macosx_10_9_intel.whl"
name, wheel, version = micropip._parse_wheel_url(url)
assert name == "scikit_learn"
assert wheel["platform"] == "macosx_10_9_intel"
def test_install_custom_url(selenium_standalone, web_server_tst_data):
server_hostname, server_port, server_log = web_server_tst_data
selenium_standalone.load_package("micropip")
selenium_standalone.run("import micropip")
base_url = f"http://{server_hostname}:{server_port}/"
url = base_url + "snowballstemmer-2.0.0-py2.py3-none-any.whl"
selenium_standalone.run(f"micropip.install('{url}')")
# wait untill micropip is loaded
time.sleep(1)
selenium_standalone.run("import snowballstemmer")
def test_last_version_from_pypi():
pytest.importorskip("distlib")
import micropip
class Namespace:
def __init__(self, **entries):
self.__dict__.update(entries)
# requirement as returned by distlib.util.parse_requirement
requirement = Namespace(
constraints=None,
extras=None,
marker=None,
name="dummy_module",
requirement="dummy_module",
url=None,
)
# available versions
versions = ["0.0.1", "0.15.5", "0.9.1"]
# building metadata as returned from
# https://pypi.org/pypi/{pkgname}/json
metadata = {
"releases": {
v: [{"filename": f"dummy_module-{v}-py3-none-any.whl"}] for v in versions
}
}
# get version number from find_wheel
wheel, ver = micropip.PACKAGE_MANAGER.find_wheel(metadata, requirement)
assert ver == "0.15.5"