bpo-31792: Restore os.environ in test_buffer when import numpy. (#4007)

This commit is contained in:
Serhiy Storchaka 2017-10-16 10:38:14 +03:00 committed by GitHub
parent 1295e11d39
commit 676db4bbf2
1 changed files with 3 additions and 2 deletions

View File

@ -17,7 +17,7 @@ from test import support
from itertools import permutations, product from itertools import permutations, product
from random import randrange, sample, choice from random import randrange, sample, choice
import warnings import warnings
import sys, array, io import sys, array, io, os
from decimal import Decimal from decimal import Decimal
from fractions import Fraction from fractions import Fraction
@ -37,7 +37,8 @@ except ImportError:
ctypes = None ctypes = None
try: try:
with warnings.catch_warnings(): with support.EnvironmentVarGuard() as os.environ, \
warnings.catch_warnings():
from numpy import ndarray as numpy_array from numpy import ndarray as numpy_array
except ImportError: except ImportError:
numpy_array = None numpy_array = None