usse/scrape/venv/lib/python3.10/site-packages/pandas/util/_tester.py

39 lines
949 B
Python
Raw Normal View History

2023-12-22 14:26:01 +00:00
"""
Entrypoint for testing from the top-level namespace.
"""
from __future__ import annotations
import os
import sys
from pandas.compat._optional import import_optional_dependency
PKG = os.path.dirname(os.path.dirname(__file__))
def test(extra_args: list[str] | None = None) -> None:
"""
Run the pandas test suite using pytest.
By default, runs with the marks --skip-slow, --skip-network, --skip-db
Parameters
----------
extra_args : list[str], default None
Extra marks to run the tests.
"""
pytest = import_optional_dependency("pytest")
import_optional_dependency("hypothesis")
cmd = ["--skip-slow", "--skip-network", "--skip-db"]
if extra_args:
if not isinstance(extra_args, list):
extra_args = [extra_args]
cmd = extra_args
cmd += [PKG]
joined = " ".join(cmd)
print(f"running: pytest {joined}")
sys.exit(pytest.main(cmd))
__all__ = ["test"]