-
Notifications
You must be signed in to change notification settings - Fork 1k
Expand file tree
/
Copy pathtest_numpy.py
More file actions
26 lines (20 loc) · 888 Bytes
/
test_numpy.py
File metadata and controls
26 lines (20 loc) · 888 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
import unittest
from distutils.version import StrictVersion
import numpy as np
import io
from contextlib import redirect_stdout
class TestNumpy(unittest.TestCase):
def test_version(self):
# b/370860329: newer versions are not capable with current tensorflow
self.assertEqual(StrictVersion(np.__version__), StrictVersion("1.26.4"))
def test_array(self):
array = np.array([1, 3])
self.assertEqual((2,), array.shape)
# Numpy must be linked to the MKL. (Occasionally, a third-party package will muck up the installation
# and numpy will be reinstalled with an OpenBLAS backing.)
def test_mkl(self):
# Fallback to check if mkl is present via show_config()
config_out = io.StringIO()
with redirect_stdout(config_out):
np.show_config()
self.assertIn("mkl-dynamic", config_out.getvalue())