def test_config_context_exception():
assert get_config()['assume_finite'] is False
try:
with config_context(assume_finite=True):
assert get_config()['assume_finite'] is True
raise ValueError()
except ValueError:
pass
assert get_config()['assume_finite'] is False
def test_config_context_exception():
assert_equal(get_config(), {'assume_finite': False})
try:
with config_context(assume_finite=True):
assert_equal(get_config(), {'assume_finite': True})
raise ValueError()
except ValueError:
pass
assert_equal(get_config(), {'assume_finite': False})
def test_config_context_exception():
assert get_config()['assume_finite'] is False
try:
with config_context(assume_finite=True):
assert get_config()['assume_finite'] is True
raise ValueError()
except ValueError:
pass
assert get_config()['assume_finite'] is False
def test_set_config():
assert get_config()['assume_finite'] is False
set_config(assume_finite=None)
assert get_config()['assume_finite'] is False
set_config(assume_finite=True)
assert get_config()['assume_finite'] is True
set_config(assume_finite=None)
assert get_config()['assume_finite'] is True
set_config(assume_finite=False)
assert get_config()['assume_finite'] is False
# No unknown arguments
assert_raises(TypeError, set_config, do_something_else=True)
def test_set_config():
assert get_config()['assume_finite'] is False
set_config(assume_finite=None)
assert get_config()['assume_finite'] is False
set_config(assume_finite=True)
assert get_config()['assume_finite'] is True
set_config(assume_finite=None)
assert get_config()['assume_finite'] is True
set_config(assume_finite=False)
assert get_config()['assume_finite'] is False
# No unknown arguments
assert_raises(TypeError, set_config, do_something_else=True)
def test_set_config():
assert_equal(get_config(), {'assume_finite': False})
set_config(assume_finite=None)
assert_equal(get_config(), {'assume_finite': False})
set_config(assume_finite=True)
assert_equal(get_config(), {'assume_finite': True})
set_config(assume_finite=None)
assert_equal(get_config(), {'assume_finite': True})
set_config(assume_finite=False)
assert_equal(get_config(), {'assume_finite': False})
# No unknown arguments
assert_raises(TypeError, set_config, do_something_else=True)
def test_set_config():
assert get_config()["assume_finite"] is False
set_config(assume_finite=None)
assert get_config()["assume_finite"] is False
set_config(assume_finite=True)
assert get_config()["assume_finite"] is True
set_config(assume_finite=None)
assert get_config()["assume_finite"] is True
set_config(assume_finite=False)
assert get_config()["assume_finite"] is False
# No unknown arguments
with pytest.raises(TypeError):
set_config(do_something_else=True)
def check_ts_X_y(model, X, y):
"""
Checks that datasets *(X, y)* was built with function
@see fn build_ts_X_y.
"""
cfg = get_config()
if cfg.get('assume_finite', True):
return # pragma: no cover
if X.dtype not in (numpy.float32, numpy.float64):
raise TypeError(
"Features must be of type float32 and float64 not {}.".format(X.dtype))
if y is not None and y.dtype not in (numpy.float32, numpy.float64):
raise TypeError( # pragma: no cover
"Features must be of type float32 and float64 not {}.".format(y.dtype))
cst = model.past
if (hasattr(model, 'preprocessing_') and model.preprocessing_ is not None):
cst += model.preprocessing_.context_length
if y is None:
if cst > 0:
raise AssertionError( # pragma: no cover
"y must be specified to give the model past data to predict, "
"it requires at least {} observations.".format(cst))
return # pragma: no cover
if y.shape[0] != X.shape[0]:
raise AssertionError( # pragma: no cover
"X and y must have the same number of rows {} != {}.".format(
X.shape[0], y.shape[0]))
if len(y.shape) > 1 and y.shape[1] != 1:
raise AssertionError( # pragma: no cover
"y must be 1-dimensional not has shape {}.".format(y.shape))
if y.shape[0] < cst:
raise AssertionError( # pragma: no cover
"y is not enough past data to predict, "
"it requires at least {} observations.".format(cst))
def __init__(self, function):
self.function = function
self.config = get_config()
update_wrapper(self, self.function)
def test_config_context():
assert get_config() == {
'assume_finite': False,
'working_memory': 1024,
'print_changed_only': False
}
# Not using as a context manager affects nothing
config_context(assume_finite=True)
assert get_config()['assume_finite'] is False
with config_context(assume_finite=True):
assert get_config() == {
'assume_finite': True,
'working_memory': 1024,
'print_changed_only': False
}
assert get_config()['assume_finite'] is False
with config_context(assume_finite=True):
with config_context(assume_finite=None):
assert get_config()['assume_finite'] is True
assert get_config()['assume_finite'] is True
with config_context(assume_finite=False):
assert get_config()['assume_finite'] is False
with config_context(assume_finite=None):
assert get_config()['assume_finite'] is False
# global setting will not be retained outside of context that
# did not modify this setting
set_config(assume_finite=True)
assert get_config()['assume_finite'] is True
assert get_config()['assume_finite'] is False
assert get_config()['assume_finite'] is True
assert get_config() == {
'assume_finite': False,
'working_memory': 1024,
'print_changed_only': False
}
# No positional arguments
assert_raises(TypeError, config_context, True)
# No unknown arguments
assert_raises(TypeError, config_context(do_something_else=True).__enter__)
def set_assume_finite(assume_finite, sleep_duration):
"""Return the value of assume_finite after waiting `sleep_duration`."""
with config_context(assume_finite=assume_finite):
time.sleep(sleep_duration)
return get_config()["assume_finite"]
def test_config_context():
assert get_config() == {
"assume_finite": False,
"working_memory": 1024,
"print_changed_only": True,
"display": "diagram",
"pairwise_dist_chunk_size": 256,
"enable_cython_pairwise_dist": True,
}
# Not using as a context manager affects nothing
config_context(assume_finite=True)
assert get_config()["assume_finite"] is False
with config_context(assume_finite=True):
assert get_config() == {
"assume_finite": True,
"working_memory": 1024,
"print_changed_only": True,
"display": "diagram",
"pairwise_dist_chunk_size": 256,
"enable_cython_pairwise_dist": True,
}
assert get_config()["assume_finite"] is False
with config_context(assume_finite=True):
with config_context(assume_finite=None):
assert get_config()["assume_finite"] is True
assert get_config()["assume_finite"] is True
with config_context(assume_finite=False):
assert get_config()["assume_finite"] is False
with config_context(assume_finite=None):
assert get_config()["assume_finite"] is False
# global setting will not be retained outside of context that
# did not modify this setting
set_config(assume_finite=True)
assert get_config()["assume_finite"] is True
assert get_config()["assume_finite"] is False
assert get_config()["assume_finite"] is True
assert get_config() == {
"assume_finite": False,
"working_memory": 1024,
"print_changed_only": True,
"display": "diagram",
"pairwise_dist_chunk_size": 256,
"enable_cython_pairwise_dist": True,
}
# No positional arguments
with pytest.raises(TypeError):
config_context(True)
# No unknown arguments
with pytest.raises(TypeError):
config_context(do_something_else=True).__enter__()
def test_config_context():
assert get_config() == {'assume_finite': False, 'working_memory': 1024}
# Not using as a context manager affects nothing
config_context(assume_finite=True)
assert get_config()['assume_finite'] is False
with config_context(assume_finite=True):
assert get_config() == {'assume_finite': True, 'working_memory': 1024}
assert get_config()['assume_finite'] is False
with config_context(assume_finite=True):
with config_context(assume_finite=None):
assert get_config()['assume_finite'] is True
assert get_config()['assume_finite'] is True
with config_context(assume_finite=False):
assert get_config()['assume_finite'] is False
with config_context(assume_finite=None):
assert get_config()['assume_finite'] is False
# global setting will not be retained outside of context that
# did not modify this setting
set_config(assume_finite=True)
assert get_config()['assume_finite'] is True
assert get_config()['assume_finite'] is False
assert get_config()['assume_finite'] is True
assert get_config() == {'assume_finite': False, 'working_memory': 1024}
# No positional arguments
assert_raises(TypeError, config_context, True)
# No unknown arguments
assert_raises(TypeError, config_context(do_something_else=True).__enter__)
os.environ["NUMEXPR_NUM_THREADS"] = "1" # export NUMEXPR_NUM_THREADS=6
from pprint import pprint
from threadpoolctl import threadpool_info, threadpool_limits
from joblib import parallel, Parallel, delayed
import numpy as np
# print(np.show_config())
import time
import pstats
from sklearn.metrics import pairwise_distances
# pprint(threadpool_info())
print('**')
print(sklearn.get_config())
def getGaussianGram(Xrow, Xcol, sigma, goFast=0):
""" get kernel (Gaussian) gram matrix
The Gram matrix K is defined as $K_ij = K(X_i , X_j) over a (sub) sample X = {X _i}, i=1,...,,n
Parameters
----------
Xrow
Xcol
sigma
goFast
Returns
-------
"""
def test_get_config_contains_sklearn_params():
skex_config = sklearnex.get_config()
sk_config = sklearn.get_config()
assert all(value in skex_config.keys() for value in sk_config.keys())
def test_config_context():
assert_equal(get_config(), {'assume_finite': False})
# Not using as a context manager affects nothing
config_context(assume_finite=True)
assert_equal(get_config(), {'assume_finite': False})
with config_context(assume_finite=True):
assert_equal(get_config(), {'assume_finite': True})
assert_equal(get_config(), {'assume_finite': False})
with config_context(assume_finite=True):
with config_context(assume_finite=None):
assert_equal(get_config(), {'assume_finite': True})
assert_equal(get_config(), {'assume_finite': True})
with config_context(assume_finite=False):
assert_equal(get_config(), {'assume_finite': False})
with config_context(assume_finite=None):
assert_equal(get_config(), {'assume_finite': False})
# global setting will not be retained outside of context that
# did not modify this setting
set_config(assume_finite=True)
assert_equal(get_config(), {'assume_finite': True})
assert_equal(get_config(), {'assume_finite': False})
assert_equal(get_config(), {'assume_finite': True})
assert_equal(get_config(), {'assume_finite': False})
# No positional arguments
assert_raises(TypeError, config_context, True)
# No unknown arguments
assert_raises(TypeError, config_context(do_something_else=True).__enter__)
