docker-python/tests/test_numpy.py at master · abhinavm24/docker-python

16 lines (12 loc) · 528 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.
        get_info("blas_mkl")

FilesExpand file tree

test_numpy.py

Latest commit

History

test_numpy.py

File metadata and controls