Disable pytest-timeout for now. (#8348)

This commit is contained in:
Jiaming Yuan
2022-10-17 23:06:10 +08:00
committed by GitHub
parent fcddbc9264
commit 2176e511fc
13 changed files with 107 additions and 58 deletions

View File

@@ -1,12 +1,15 @@
import sys
from hypothesis import strategies, given, settings, assume, note
import pytest
from hypothesis import assume, given, note, settings, strategies
import xgboost as xgb
from xgboost import testing
sys.path.append("tests/python")
import testing as tm
pytestmark = pytest.mark.timeout(10)
pytestmark = testing.timeout(10)
parameter_strategy = strategies.fixed_dictionaries({
'booster': strategies.just('gblinear'),

View File

@@ -1,21 +1,23 @@
'''Test model IO with pickle.'''
import pickle
import numpy as np
import subprocess
import os
import sys
import json
import os
import pickle
import subprocess
import sys
import numpy as np
import pytest
import xgboost as xgb
from xgboost import XGBClassifier
from xgboost import XGBClassifier, testing
sys.path.append("tests/python")
import testing as tm
model_path = './model.pkl'
pytestmark = testing.timeout(30)
pytestmark = pytest.mark.timeout(30)
def build_dataset():
N = 10

View File

@@ -1,11 +1,12 @@
import sys
import pytest
import numpy as np
import xgboost as xgb
import pytest
from hypothesis import assume, given, settings, strategies
from xgboost.compat import PANDAS_INSTALLED
from hypothesis import given, strategies, assume, settings
import xgboost as xgb
from xgboost import testing
if PANDAS_INSTALLED:
from hypothesis.extra.pandas import column, data_frames, range_indexes
@@ -16,8 +17,8 @@ else:
sys.path.append("tests/python")
import testing as tm
from test_predict import run_predict_leaf # noqa
from test_predict import run_threaded_predict # noqa
from test_predict import run_predict_leaf # noqa
rng = np.random.RandomState(1994)
@@ -32,7 +33,8 @@ predict_parameter_strategy = strategies.fixed_dictionaries({
'num_parallel_tree': strategies.sampled_from([1, 4]),
})
pytestmark = pytest.mark.timeout(20)
pytestmark = testing.timeout(20)
class TestGPUPredict:
def test_predict(self):

View File

@@ -1,17 +1,21 @@
import numpy as np
import xgboost
import os
import itertools
import os
import shutil
import sys
import urllib.request
import zipfile
import sys
import pytest
import numpy as np
import xgboost
from xgboost import testing
sys.path.append("tests/python")
import testing as tm # noqa
import testing as tm # noqa
pytestmark = testing.timeout(10)
pytestmark = pytest.mark.timeout(10)
class TestRanking:
@classmethod

View File

@@ -1,15 +1,18 @@
from typing import Dict, Any
import numpy as np
import sys
from typing import Any, Dict
import numpy as np
import pytest
from hypothesis import assume, given, note, settings, strategies
import xgboost as xgb
from hypothesis import given, strategies, assume, settings, note
from xgboost import testing
sys.path.append("tests/python")
import testing as tm
import test_updaters as test_up
import testing as tm
pytestmark = pytest.mark.timeout(30)
pytestmark = testing.timeout(30)
parameter_strategy = strategies.fixed_dictionaries({
'max_depth': strategies.integers(0, 11),