def func(env): global counter with env: nthreads = os.getenv('OMP_NUM_THREADS') expected = omp_num_threads() with pool_threading() as pool: assert_equal(int(os.environ['OMP_NUM_THREADS']), 1) if mkl is not None: assert_equal(mkl.get_max_threads(), 1) counter = 0 pool.map(func_thread, range(pool._processes)) assert_equal(os.getenv('OMP_NUM_THREADS'), nthreads) if mkl is not None: assert_equal(mkl.get_max_threads(), mkl_nthreads) assert_equal(counter, expected) assert_not_in('OMP_NUM_THREADS', os.environ)
def func(env): global counter with env: nthreads = os.getenv("OMP_NUM_THREADS") expected = omp_num_threads() with pool_threading() as pool: assert_equal(int(os.environ["OMP_NUM_THREADS"]), 1) if mkl is not None: assert_equal(mkl.get_max_threads(), 1) counter = 0 pool.map(func_thread, range(pool._processes)) assert_equal(os.getenv("OMP_NUM_THREADS"), nthreads) if mkl is not None: assert_equal(mkl.get_max_threads(), mkl_nthreads) assert_equal(counter, expected) assert_not_in("OMP_NUM_THREADS", os.environ)
def test_homothety_operator(): s = HomothetyOperator(1) assert s.C is s.T is s.H is s.I is s s = HomothetyOperator(-1) assert s.C is s.T is s.H is s.I is s s = HomothetyOperator(2.) assert s.C is s.T is s.H is s assert_is_not(s.I, s) def func(o): assert_is_instance(o, HomothetyOperator) for o in (s.I, s.I.C, s.I.T, s.I.H, s.I.I): yield func, o s = HomothetyOperator(complex(1, 1)) assert_is(s.T, s) assert_is(s.H, s.C) assert_not_in(s.I, (s, s.C)) assert_not_in(s.I.C, (s, s.C)) assert_is_instance(s.C, HomothetyOperator) for o in (s.I, s.I.C, s.I.T, s.I.H, s.I.I): yield func, o