docker-python/tests/test_numpy.py at master · AIComputerVision/docker-python · GitHub

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")

Expand file tree