Move Python testing utilities into xgboost module. (#8379)

- Add typehints.
- Fixes for pylint.

Co-authored-by: Hyunsu Philip Cho <chohyu01@cs.washington.edu>
This commit is contained in:
Jiaming Yuan
2022-10-26 16:56:11 +08:00
committed by GitHub
parent 7e53189e7c
commit cf70864fa3
66 changed files with 652 additions and 595 deletions

View File

@@ -4,12 +4,11 @@ import tempfile
import numpy as np
import pytest
import testing as tm
import xgboost as xgb
from xgboost import testing
from xgboost import testing as tm
pytestmark = testing.timeout(10)
pytestmark = tm.timeout(10)
class TestOMP:
@@ -86,7 +85,7 @@ class TestOMP:
def test_with_omp_thread_limit(self):
args = [
"python", os.path.join(
tm.PROJECT_ROOT, "tests", "python", "with_omp_limit.py"
os.path.dirname(tm.normpath(__file__)), "with_omp_limit.py"
)
]
results = []