def test_basic_centos(self): """Default mkl building block""" m = mkl(eula=True) self.assertEqual( str(m), r'''# MKL version 2018.3-051 RUN rpm --import https://yum.repos.intel.com/intel-gpg-keys/GPG-PUB-KEY-INTEL-SW-PRODUCTS-2019.PUB && \ yum-config-manager --add-repo https://yum.repos.intel.com/mkl/setup/intel-mkl.repo && \ yum install -y \ intel-mkl-64bit-2018.3-051 && \ rm -rf /var/cache/yum/* RUN echo "source /opt/intel/mkl/bin/mklvars.sh intel64" >> /etc/bashrc''')
def test_version(self): """Version option""" m = mkl(eula=True, version='2018.2-046') self.assertEqual( str(m), r'''# MKL version 2018.2-046 RUN apt-get update -y && \ apt-get install -y --no-install-recommends \ apt-transport-https \ ca-certificates \ wget && \ rm -rf /var/lib/apt/lists/* RUN wget -qO - https://apt.repos.intel.com/intel-gpg-keys/GPG-PUB-KEY-INTEL-SW-PRODUCTS-2019.PUB | apt-key add - && \ echo "deb https://apt.repos.intel.com/mkl all main" >> /etc/apt/sources.list.d/hpccm.list && \ apt-get update -y && \ apt-get install -y --no-install-recommends \ intel-mkl-64bit-2018.2-046 && \ rm -rf /var/lib/apt/lists/* RUN echo "source /opt/intel/mkl/bin/mklvars.sh intel64" >> /etc/bash.bashrc''')
def test_mklvars(self): """mklvars is False""" m = mkl(eula=True, mklvars=False) self.maxDiff = None self.assertEqual( str(m), r'''# MKL version 2018.3-051 RUN apt-get update -y && \ apt-get install -y --no-install-recommends \ apt-transport-https \ ca-certificates \ wget && \ rm -rf /var/lib/apt/lists/* RUN wget -qO - https://apt.repos.intel.com/intel-gpg-keys/GPG-PUB-KEY-INTEL-SW-PRODUCTS-2019.PUB | apt-key add - && \ echo "deb https://apt.repos.intel.com/mkl all main" >> /etc/apt/sources.list.d/hpccm.list && \ apt-get update -y && \ apt-get install -y --no-install-recommends \ intel-mkl-64bit-2018.3-051 && \ rm -rf /var/lib/apt/lists/* ENV CPATH=/opt/intel/mkl/include:$CPATH \ LD_LIBRARY_PATH=/opt/intel/mkl/lib/intel64:/opt/intel/lib/intel64:$LD_LIBRARY_PATH \ LIBRARY_PATH=/opt/intel/mkl/lib/intel64:/opt/intel/lib/intel64:$LIBRARY_PATH \ MKLROOT=/opt/intel/mkl''')
def test_defaults(self): """Default mkl building block, no eula agreement""" m = mkl() with self.assertRaises(RuntimeError): str(m)