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_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__)