def pytest_runtest_setup(item):
global _sort_on_spark
global _sort_locally
order = item.get_closest_marker('ignore_order')
if order:
if order.kwargs.get('local', False):
_sort_on_spark = False
_sort_locally = True
else:
_sort_on_spark = True
_sort_locally = False
else:
_sort_on_spark = False
_sort_locally = False
global _incompat
if item.get_closest_marker('incompat'):
_incompat = True
else:
_incompat = False
global _approximate_float_args
app_f = item.get_closest_marker('approximate_float')
if app_f:
_approximate_float_args = app_f.kwargs
else:
_approximate_float_args = None
global _allow_any_non_gpu
global _non_gpu_allowed
non_gpu = item.get_closest_marker('allow_non_gpu')
if non_gpu:
if non_gpu.kwargs and non_gpu.kwargs['any']:
_allow_any_non_gpu = True
_non_gpu_allowed = []
elif non_gpu.args:
_allow_any_non_gpu = False
_non_gpu_allowed = non_gpu.args
else:
pytest.warn('allow_non_gpu marker without anything allowed')
_allow_any_non_gpu = False
_non_gpu_allowed = []
else:
_allow_any_non_gpu = False
_non_gpu_allowed = []
global _limit
limit_mrk = item.get_closest_marker('limit')
if limit_mrk:
_limit = _get_limit_from_mark(limit_mrk)
else:
_limit = -1
def pytest_runtest_setup(item):
global _sort_on_spark
global _sort_locally
order = item.get_closest_marker('ignore_order')
if order:
if order.kwargs.get('local', False):
_sort_on_spark = False
_sort_locally = True
else:
_sort_on_spark = True
_sort_locally = False
else:
_sort_on_spark = False
_sort_locally = False
global _incompat
if item.get_closest_marker('incompat'):
_incompat = True
else:
_incompat = False
global _approximate_float_args
app_f = item.get_closest_marker('approximate_float')
if app_f:
_approximate_float_args = app_f.kwargs
else:
_approximate_float_args = None
global _allow_any_non_gpu
global _non_gpu_allowed
_non_gpu_allowed_databricks = []
_allow_any_non_gpu_databricks = False
non_gpu_databricks = item.get_closest_marker('allow_non_gpu_databricks')
non_gpu = item.get_closest_marker('allow_non_gpu')
if non_gpu_databricks:
if is_databricks_runtime():
if non_gpu_databricks.kwargs and non_gpu_databricks.kwargs['any']:
_allow_any_non_gpu_databricks = True
elif non_gpu_databricks.args:
_non_gpu_allowed_databricks = non_gpu_databricks.args
else:
pytest.warn(
'allow_non_gpu_databricks marker without anything allowed')
if non_gpu:
if non_gpu.kwargs and non_gpu.kwargs['any']:
_allow_any_non_gpu = True
_non_gpu_allowed = []
elif non_gpu.args:
_allow_any_non_gpu = False
_non_gpu_allowed = non_gpu.args
else:
pytest.warn('allow_non_gpu marker without anything allowed')
_allow_any_non_gpu = False
_non_gpu_allowed = []
else:
_allow_any_non_gpu = False
_non_gpu_allowed = []
_allow_any_non_gpu = _allow_any_non_gpu | _allow_any_non_gpu_databricks
if _non_gpu_allowed and _non_gpu_allowed_databricks:
_non_gpu_allowed = _non_gpu_allowed + _non_gpu_allowed_databricks
elif _non_gpu_allowed_databricks:
_non_gpu_allowed = _non_gpu_allowed_databricks
global _validate_execs_in_gpu_plan
validate_execs = item.get_closest_marker('validate_execs_in_gpu_plan')
if validate_execs and validate_execs.args:
_validate_execs_in_gpu_plan = validate_execs.args
else:
_validate_execs_in_gpu_plan = []
global _limit
limit_mrk = item.get_closest_marker('limit')
if limit_mrk:
_limit = _get_limit_from_mark(limit_mrk)
else:
_limit = -1
