e413b037182d131100ad48ec157f155943f4a4db,pyemma/coordinates/tests/test_assign.py,TestClusterAssign,test_threads_omp_env_arg_borked,#TestClusterAssign#,219

Before Change

    def test_threads_omp_env_arg_borked(self):
        import os
        os.environ["OMP_NUM_THREADS"] = "this is not right"
        try:
            import psutil
            X = np.random.random((1000, 3))
            centers = X[np.random.choice(1000, 10)]
            // note: we want another job number here, but it will be ignored!
            res = coor.assign_to_centers(X, centers, n_jobs=None, return_dtrajs=False)
            self.assertEqual(res.n_jobs, psutil.cpu_count())
        finally:
            del os.environ["OMP_NUM_THREADS"]

    def test_threads_cpu_count_def_arg(self):
        import psutil
        X = np.random.random((1000, 3))
        centers = X[np.random.choice(1000, 10)]

After Change

    def test_threads_omp_env_arg_borked(self):
         if the env var can not be interpreted as int, fall back to one thread. 
        expected = 3
        with patch("psutil.cpu_count", lambda: expected), temporary_env("OMP_NUM_THREADS", "this is not right"):
            res = coor.assign_to_centers(self.X, self.centers_big, n_jobs=None, return_dtrajs=False)
            self.assertEqual(res.n_jobs, expected)

Instances