mirror of https://github.com/pyodide/pyodide.git
28 lines
735 B
Python
28 lines
735 B
Python
import pytest
|
|
|
|
|
|
def test_scikit_learn(selenium_standalone, request):
|
|
selenium = selenium_standalone
|
|
if selenium.browser == "chrome":
|
|
request.applymarker(pytest.mark.xfail(run=False, reason="chrome not supported"))
|
|
selenium.load_package("scikit-learn")
|
|
assert (
|
|
selenium.run(
|
|
"""
|
|
import numpy as np
|
|
import sklearn
|
|
from sklearn.linear_model import LogisticRegression
|
|
|
|
rng = np.random.RandomState(42)
|
|
X = rng.rand(100, 20)
|
|
y = rng.randint(5, size=100)
|
|
|
|
estimator = LogisticRegression(solver='liblinear')
|
|
estimator.fit(X, y)
|
|
print(estimator.predict(X))
|
|
estimator.score(X, y)
|
|
"""
|
|
)
|
|
> 0
|
|
)
|