import pytest
from pytest_pyodide import run_in_pyodide


@pytest.mark.driver_timeout(40)
@run_in_pyodide(packages=["scipy"])
def test_scipy_linalg(selenium):
    import numpy as np
    import scipy.linalg
    from numpy.testing import assert_allclose

    N = 10
    X = np.random.RandomState(42).rand(N, N)

    X_inv = scipy.linalg.inv(X)

    res = X.dot(X_inv)

    assert_allclose(res, np.identity(N), rtol=1e-07, atol=1e-9)


@pytest.mark.driver_timeout(40)
@run_in_pyodide(packages=["scipy"])
def test_brentq(selenium):
    from scipy.optimize import brentq

    brentq(lambda x: x, -1, 1)


@pytest.mark.driver_timeout(40)
@run_in_pyodide(packages=["scipy"])
def test_dlamch(selenium):
    from scipy.linalg import lapack

    lapack.dlamch("Epsilon-Machine")


@pytest.mark.driver_timeout(40)
@run_in_pyodide(packages=["scipy"])
def test_binom_ppf(selenium):
    from scipy.stats import binom

    assert binom.ppf(0.9, 1000, 0.1) == 112


@pytest.mark.driver_timeout(40)
@run_in_pyodide(packages=["pytest", "scipy-tests"])
def test_scipy_pytest(selenium):
    import pytest

    def runtest(module, filter):
        pytest.main(
            [
                "--pyargs",
                f"scipy.{module}",
                "--continue-on-collection-errors",
                "-vv",
                "-k",
                filter,
            ]
        )

    runtest("odr", "explicit")
    runtest("signal.tests.test_ltisys", "TestImpulse2")
    runtest("stats.tests.test_multivariate", "haar")
    runtest("sparse.linalg._eigen", "test_svds_parameter_k_which")


@pytest.mark.driver_timeout(40)
@run_in_pyodide(packages=["scipy"])
def test_cpp_exceptions(selenium):
    import numpy as np
    import pytest
    from scipy.spatial.distance import cdist

    out = np.ones((2, 2))
    arr = np.array([[1, 2]])

    with pytest.raises(ValueError, match="Output array has incorrect shape"):
        cdist(arr, arr, out=out)
    from scipy.sparse._sparsetools import test_throw_error

    with pytest.raises(MemoryError):
        test_throw_error()
    from scipy.signal import lombscargle

    with pytest.raises(ValueError):
        lombscargle(x=[1], y=[1, 2], freqs=[1, 2, 3])