docker-python/tests/test_numpy.py at main · MunirAbobaker/docker-python

16 lines (12 loc) · 569 Bytes

import unittest
import numpy as np
from numpy.distutils.system_info import get_info
class TestNumpy(unittest.TestCase):    
    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):
        # This will throw an exception if the MKL is not linked correctly or return an empty dict.
        self.assertTrue(get_info("blas_mkl"))

FilesExpand file tree

test_numpy.py

Latest commit

History

test_numpy.py

File metadata and controls