change docker install script (#3524)

conda/tvm/meta.yaml

@@ -50,12 +50,12 @@ test:
     - topi
     - nnvm
   requires:
-    - nose
+    - pytest
     - scipy
   source_files:
     - tests/python
   commands:
-    - python -m nose -v tests/python/integration
+    - python -m pytest -v tests/python/integration
 
 about:
   home: https://github.com/dmlc/tvm

docker/Dockerfile.conda_cpu

@@ -30,9 +30,9 @@ RUN curl -o ~/miniconda.sh -O https://repo.continuum.io/miniconda/Miniconda3-lat
 # Cache some of the packages for the builds
 RUN /opt/conda/bin/conda install --download-only cmake make zlib && \
     /opt/conda/bin/conda install --download-only -c numba llvmdev=8.0.0 && \
-    /opt/conda/bin/conda create -n py35 --download-only nose scipy numpy=1.11 cython decorator python=3.5 && \
-    /opt/conda/bin/conda create -n py36 --download-only nose scipy numpy=1.11 cython decorator python=3.6 && \
-    /opt/conda/bin/conda create -n py37 --download-only nose scipy numpy=1.11 cython decorator python=3.7
+    /opt/conda/bin/conda create -n py35 --download-only pytest scipy numpy=1.11 cython decorator python=3.5 && \
+    /opt/conda/bin/conda create -n py36 --download-only pytest scipy numpy=1.11 cython decorator python=3.6 && \
+    /opt/conda/bin/conda create -n py37 --download-only pytest scipy numpy=1.11 cython decorator python=3.7
 
 ENV PATH /opt/conda/bin:$PATH
 ENV CONDA_BLD_PATH /tmp

docker/Dockerfile.demo_cpu

@@ -26,7 +26,7 @@ RUN bash /install/ubuntu_install_core.sh
 
 # Python: basic dependencies
 RUN apt-get update && apt-get install -y python3-dev python3-pip
-RUN pip3 install numpy nose-timer cython decorator scipy
+RUN pip3 install numpy pytest cython decorator scipy
 
 # LLVM
 RUN echo deb http://apt.llvm.org/xenial/ llvm-toolchain-xenial-6.0 main \

docker/Dockerfile.demo_gpu

@@ -26,7 +26,7 @@ RUN bash /install/ubuntu_install_core.sh
 
 # Python: basic dependencies
 RUN apt-get update && apt-get install -y python3-dev python3-pip
-RUN pip3 install numpy nose-timer cython decorator scipy
+RUN pip3 install numpy pytest cython decorator scipy
 
 # LLVM
 RUN echo deb http://apt.llvm.org/xenial/ llvm-toolchain-xenial-6.0 main \

docker/Dockerfile.demo_opencl

@@ -42,7 +42,7 @@ RUN apt-get install -y libopenblas-dev
 
 RUN echo "Installing Python"
 RUN apt-get install -y python3-dev python3-pip
-RUN pip3 install setuptools numpy nose-timer cython decorator scipy tornado psutil xgboost
+RUN pip3 install setuptools numpy pytest cython decorator scipy tornado psutil xgboost
 
 RUN echo "Installing Jupyter notebook"
 RUN pip3 install matplotlib Image Pillow jupyter[notebook]

docker/install/ubuntu_install_python_package.sh

@@ -6,9 +6,9 @@
 # to you under the Apache License, Version 2.0 (the
 # "License"); you may not use this file except in compliance
 # with the License.  You may obtain a copy of the License at
-# 
+#
 #   http://www.apache.org/licenses/LICENSE-2.0
-# 
+#
 # Unless required by applicable law or agreed to in writing,
 # software distributed under the License is distributed on an
 # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
@@ -21,5 +21,4 @@ set -u
 set -o pipefail
 
 # install libraries for python package on ubuntu
-pip2 install nose pylint==1.9.4 six numpy nose-timer cython decorator scipy tornado typing antlr4-python2-runtime attrs packaging
-pip3 install nose pylint==1.9.4 six numpy nose-timer cython decorator scipy tornado typed_ast pytest mypy orderedset antlr4-python3-runtime attrs requests Pillow packaging
+pip3 install pylint==1.9.4 six numpy pytest cython decorator scipy tornado typed_ast pytest mypy orderedset antlr4-python3-runtime attrs requests Pillow packaging

docs/contribute/pull_request.rst

@@ -96,7 +96,7 @@ Necessary dependencies:
 
 .. code:: bash
 
-  pip install --user nose Cython
+  pip install --user pytest Cython
 
 If you want to run all tests:
 
@@ -118,4 +118,4 @@ If you want to run a single test:
   export PYTHONPATH=python:topi/python
   rm -rf python/tvm/*.pyc python/tvm/*/*.pyc python/tvm/*/*/*.pyc
 
-  TVM_FFI=ctypes python -m nose -v tests/python/unittest/test_pass_storage_rewrite.py
+  TVM_FFI=ctypes python -m pytest -v tests/python/unittest/test_pass_storage_rewrite.py

nnvm/tests/python/frontend/mxnet/test_forward.py

@@ -31,7 +31,7 @@ import model_zoo
 
 def verify_mxnet_frontend_impl(mx_symbol, data_shape=(1, 3, 224, 224), out_shape=(1, 1000),
                                gluon_impl=False, name=None, dtype='float32'):
-    """Use name different from test to avoid let nose pick it up"""
+    """Use name different from test to avoid pytest picking it up"""
     if gluon_impl:
         def get_gluon_output(name, x):
             net = vision.get_model(name)

tests/python/contrib/test_mxnet_bridge.py

@@ -17,7 +17,7 @@
 def mxnet_check():
     """This is a simple test function for MXNet bridge
 
-    It is not included as nosetests, because of its dependency on mxnet
+    It is not included as pytests, because of its dependency on mxnet
 
     User can directly run this script to verify correctness.
     """

tests/python/contrib/test_nnpack.py

@@ -18,7 +18,7 @@ import tvm
 import numpy as np
 import scipy.signal
 from tvm.contrib import nnpack
-from nose import SkipTest
+import pytest
 
 
 def test_fully_connected_inference():
@@ -34,11 +34,11 @@ def test_fully_connected_inference():
 
     def verify(target="llvm"):
         if not tvm.module.enabled(target):
-            raise SkipTest("skip because %s is not enabled..." % target)
+            pytest.skip("%s is not enabled..." % target)
         if not tvm.get_global_func("tvm.contrib.nnpack.fully_connected_inference", True):
-            raise SkipTest("skip because extern function is not available")
+            pytest.skip("extern function is not available")
         if not nnpack.is_available():
-            raise SkipTest("skip because nnpack is not available")
+            pytest.skip("nnpack is not available")
 
         ctx = tvm.cpu(0)
         f = tvm.build(s, [A, B, D, bias], target)
@@ -112,11 +112,11 @@ def test_convolution_inference():
                algorithm=nnpack.ConvolutionAlgorithm.AUTO,
                with_bias=True):
         if not tvm.module.enabled(target):
-            raise SkipTest("skip because %s is not enabled..." % target)
+            pytest.skip("%s is not enabled..." % target)
         if not tvm.get_global_func("tvm.contrib.nnpack.fully_connected_inference", True):
-            raise SkipTest("skip because extern function is not available")
+            pytest.skip("extern function is not available")
         if not nnpack.is_available():
-            raise SkipTest("skip because nnpack is not available")
+            pytest.skip("nnpack is not available")
 
         ctx = tvm.cpu(0)
         output = nnpack.convolution_inference(
@@ -174,11 +174,11 @@ def test_convolution_inference_without_weight_transform():
                algorithm=nnpack.ConvolutionAlgorithm.AUTO,
                with_bias=True):
         if not tvm.module.enabled(target):
-            raise SkipTest("skip because %s is not enabled..." % target)
+            pytest.skip("%s is not enabled..." % target)
         if not tvm.get_global_func("tvm.contrib.nnpack.fully_connected_inference", True):
-            raise SkipTest("skip because extern function is not available")
+            pytest.skip("extern function is not available")
         if not nnpack.is_available():
-            raise SkipTest("skip because nnpack is not available")
+            pytest.skip("nnpack is not available")
 
         ctx = tvm.cpu(0)
         transformed_kernel = nnpack.convolution_inference_weight_transform(
@@ -209,5 +209,4 @@ def test_convolution_inference_without_weight_transform():
 
 
 if __name__ == "__main__":
-    import nose
-    nose.runmodule()
+    pytest.main()

tests/python/contrib/test_rpc_proxy.py

@@ -24,7 +24,7 @@ from tvm import rpc
 def rpc_proxy_check():
     """This is a simple test function for RPC Proxy
 
-    It is not included as nosetests, because:
+    It is not included as pytests, because:
     - It depends on tornado
     - It relies on the fact that Proxy starts before client and server connects,
       which is often the case but not always

tests/python/frontend/mxnet/test_forward.py

@@ -34,7 +34,7 @@ def verify_mxnet_frontend_impl(mx_symbol,
                                gluon_impl=False,
                                name=None,
                                dtype='float32'):
-    """Use name different from test to avoid let nose pick it up"""
+    """Use name different from test to avoid pytest picking it up"""
     if gluon_impl:
         def get_gluon_output(name, x):
             net = vision.get_model(name)

tests/python/integration/test_winograd_nnpack.py

@@ -23,7 +23,7 @@ from tvm.contrib.pickle_memoize import memoize
 import topi
 import topi.testing
 from topi.util import get_const_tuple
-from nose import SkipTest
+from pytest import skip
 
 
 def verify_conv2d_nchw(batch, in_channel, in_size, num_filter, kernel, stride, padding, dilation=1, add_bias=False, add_relu=False,
@@ -60,7 +60,7 @@ def verify_conv2d_nchw(batch, in_channel, in_size, num_filter, kernel, stride, p
     def check_device(device):
         ctx = tvm.context(device, 0)
         if not ctx.exist:
-            raise SkipTest("Skip because %s is not enabled" % device)
+            skip("s is not enabled" % device)
         print("Running on target: %s" % device)
         with tvm.target.create(device):
             C = topi.nn.conv2d(A, W, stride, padding, dilation, layout='NCHW', out_dtype=dtype)
@@ -99,10 +99,10 @@ class WinogradFallback(autotvm.FallbackContext):
 
 def test_conv2d_nchw():
     if not tvm.get_global_func("tvm.contrib.nnpack.convolution_inference_without_weight_transform", True):
-        raise SkipTest("skip because extern function is not available")
+        skip("extern function is not available")
 
     if not nnpack.is_available():
-        raise SkipTest("skip because nnpack is not available")
+        skip("nnpack is not available")
 
     devices = ['llvm -device=arm_cpu']
     autotvm.DispatchContext.current.silent = True
@@ -139,5 +139,5 @@ def test_conv2d_nchw():
 
 
 if __name__ == "__main__":
-    import nose
-    nose.runmodule()
+    import pytest
+    pytest.main()

tests/python/relay/test_ir_parser.py

@@ -17,11 +17,12 @@
 import tvm
 from tvm import relay
 from tvm.relay.analysis import graph_equal, assert_graph_equal
-from nose.tools import nottest, raises
+from tvm.relay.analysis import alpha_equal, assert_alpha_equal
+import pytest
 from numpy import isclose
 from typing import Union
 from functools import wraps
-raises_parse_error = raises(tvm._ffi.base.TVMError)
+raises_parse_error = pytest.mark.xfail(raises=tvm._ffi.base.TVMError)
 
 SEMVER = "v0.0.4"
 
@@ -186,7 +187,7 @@ def test_op_assoc():
     assert graph_equal(parse_text("1 == 1 < 1 + 1 * 1"), parse_text("1 == (1 < (1 + (1 * 1)))"))
 
 
-@nottest
+@pytest.mark.skip
 def test_vars():
     # temp vars won't work b/c they start with a digit
     # # temp var
@@ -538,11 +539,6 @@ def test_builtin_types():
         parse_text("let %_ : {} = (); ()".format(builtin_type))
 
 
-@nottest
-def test_call_type():
-    assert False
-
-
 def test_tensor_type():
     assert parses_as(
         "let %_ : Tensor[(), float32] = (); ()",

tests/python/relay/test_op_level3.py

@@ -17,7 +17,7 @@
 """ Support level3 operator test cases.
 """
 import numpy as np
-from nose.tools import raises
+import pytest
 import tvm
 from tvm import relay
 from tvm.relay import create_executor, transform
@@ -220,8 +220,7 @@ def test_squeeze_infer_type():
     assert yy.checked_type == relay.TensorType(
         (4,), "float32")
 
-
-@raises(tvm._ffi.base.TVMError)
+@pytest.mark.xfail(raises=tvm._ffi.base.TVMError)
 def test_squeeze_bad_axes_infer_type():
     n, t, d = 1, 4, 1
     x = relay.var("x", relay.TensorType((n, t, d), "float32"))

tests/python/relay/test_pass_check_kind.py

@@ -17,8 +17,7 @@
 import tvm
 from tvm import relay
 from tvm.relay.analysis import check_kind
-from nose.tools import raises
-
+import pytest
 
 def test_typevar_kind():
     # returns the same kind
@@ -111,7 +110,7 @@ def test_typecall_kind():
     assert check_kind(call, new_mod) == relay.Kind.Type
 
 
-@raises(tvm._ffi.base.TVMError)
+@pytest.mark.xfail(raises=tvm._ffi.base.TVMError)
 def test_invalid_tuple_kind():
     tp1 = relay.TypeVar('tp1', relay.Kind.Shape)
     tp2 = relay.TypeVar('tp2', relay.Kind.BaseType)
@@ -122,7 +121,7 @@ def test_invalid_tuple_kind():
     check_kind(tup_ty)
 
 
-@raises(tvm._ffi.base.TVMError)
+@pytest.mark.xfail(raises=tvm._ffi.base.TVMError)
 def test_invalid_func_kind():
     tp1 = relay.TypeVar('tp1', relay.Kind.Shape)
     tp2 = relay.TypeVar('tp2', relay.Kind.BaseType)
@@ -137,14 +136,14 @@ def test_invalid_func_kind():
     check_kind(tf)
 
 
-@raises(tvm._ffi.base.TVMError)
+@pytest.mark.xfail(raises=tvm._ffi.base.TVMError)
 def test_invalid_ref_kind():
     tp = relay.TypeVar('tp', relay.Kind.Shape)
     rt = relay.RefType(tp)
     check_kind(rt)
 
 
-@raises(tvm._ffi.base.TVMError)
+@pytest.mark.xfail(raises=tvm._ffi.base.TVMError)
 def test_invalid_relation_kind():
     tp1 = relay.TypeVar('tp1', relay.Kind.Shape)
     tp2 = relay.TypeVar('tp2', relay.Kind.BaseType)
@@ -156,14 +155,14 @@ def test_invalid_relation_kind():
     check_kind(tr)
 
 
-@raises(tvm._ffi.base.TVMError)
+@pytest.mark.xfail(raises=tvm._ffi.base.TVMError)
 def test_typecall_invalid_callee():
     # global type var must be an ADT handle
     gtv = relay.GlobalTypeVar('v1', relay.Kind.Type)
     check_kind(relay.TypeCall(gtv, []))
 
 
-@raises(tvm._ffi.base.TVMError)
+@pytest.mark.xfail(raises=tvm._ffi.base.TVMError)
 def test_typecall_invalid_args():
     # args must all be type kind
     mod = relay.Module()
@@ -174,7 +173,7 @@ def test_typecall_invalid_args():
     check_kind(relay.TypeCall(gtv, [data]))
 
 
-@raises(tvm._ffi.base.TVMError)
+@pytest.mark.xfail(raises=tvm._ffi.base.TVMError)
 def test_typecall_invalid_num_args():
     mod = relay.Module()
     gtv = relay.GlobalTypeVar('v1')
@@ -184,7 +183,7 @@ def test_typecall_invalid_num_args():
     check_kind(relay.TypeCall(gtv, []))
 
 
-@raises(tvm._ffi.base.TVMError)
+@pytest.mark.xfail(raises=tvm._ffi.base.TVMError)
 def test_func_with_invalid_ret_type():
     tp1 = relay.TypeVar('tp1', relay.Kind.Type)
     tp2 = relay.TypeVar('tp2', relay.Kind.Shape)
@@ -193,7 +192,7 @@ def test_func_with_invalid_ret_type():
     check_kind(tf)
 
 
-@raises(tvm._ffi.base.TVMError)
+@pytest.mark.xfail(raises=tvm._ffi.base.TVMError)
 def test_func_with_invalid_arg_types():
     tp1 = relay.TypeVar('tp1', relay.Kind.Shape)
     tp2 = relay.TypeVar('tp2', relay.Kind.Type)
@@ -202,7 +201,7 @@ def test_func_with_invalid_arg_types():
     check_kind(tf)
 
 
-@raises(tvm._ffi.base.TVMError)
+@pytest.mark.xfail(raises=tvm._ffi.base.TVMError)
 def test_func_with_invalid_tuple():
     tp1 = relay.TypeVar('tp1', relay.Kind.Shape)
 
@@ -212,7 +211,7 @@ def test_func_with_invalid_tuple():
     check_kind(tf)
 
 
-@raises(tvm._ffi.base.TVMError)
+@pytest.mark.xfail(raises=tvm._ffi.base.TVMError)
 def test_func_with_invalid_relation():
     tp1 = relay.TypeVar('tp1', relay.Kind.Type)
     tp2 = relay.TypeVar('tp2', relay.Kind.Shape)
@@ -225,7 +224,7 @@ def test_func_with_invalid_relation():
     check_kind(tf)
 
 
-@raises(tvm._ffi.base.TVMError)
+@pytest.mark.xfail(raises=tvm._ffi.base.TVMError)
 def test_tuple_with_invalid_func():
     tensor_type = relay.TensorType(tvm.convert([1, 2, 3]), 'float32')
 

tests/python/relay/test_pass_dead_code_elimination.py

@@ -14,8 +14,6 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
-from nose.tools import nottest
-
 import tvm
 from tvm import relay
 from tvm.relay import Function, transform

tests/python/relay/test_pass_manager.py

@@ -16,6 +16,7 @@
 # under the License.
 """Unit tests for relay pass manager."""
 import numpy as np
+import pytest
 
 import tvm
 from tvm import relay
@@ -504,7 +505,7 @@ def test_sequential_with_scoping():
     assert analysis.alpha_equal(zz, zexpected)
 
 
-def test_print_ir():
+def test_print_ir(capfd):
     shape = (1, 2, 3)
     tp = relay.TensorType(shape, "float32")
     x = relay.var("x", tp)
@@ -519,53 +520,15 @@ def test_print_ir():
         relay.transform.DeadCodeElimination()
     ])
 
-    def redirect_output(call):
-        """Redirect the C++ logging info."""
-        import sys
-        import os
-        import threading
-        stderr_fileno = sys.stderr.fileno()
-        stderr_save = os.dup(stderr_fileno)
-        stderr_pipe = os.pipe()
-        os.dup2(stderr_pipe[1], stderr_fileno)
-        os.close(stderr_pipe[1])
-        output = ''
-
-        def record():
-            nonlocal output
-            while True:
-                data = os.read(stderr_pipe[0], 1024)
-                if not data:
-                    break
-                output += data.decode("utf-8")
-
-        t = threading.Thread(target=record)
-        t.start()
-        call()
-        os.close(stderr_fileno)
-        t.join()
-        os.close(stderr_pipe[0])
-        os.dup2(stderr_save, stderr_fileno)
-        os.close(stderr_save)
-
-        return output
-
-    def run_pass():
-        mod = relay.Module({"main": func})
-        with relay.build_config(opt_level=3):
-            mod = seq(mod)
+    mod = relay.Module({"main": func})
+    with relay.build_config(opt_level=3):
+        mod = seq(mod)
+
+    out = capfd.readouterr().err
 
-    out = redirect_output(run_pass)
     assert "Dumping the module IR" in out
     assert "multiply" in out
 
 
 if __name__ == "__main__":
-    test_function_class_pass()
-    test_module_class_pass()
-    test_module_pass()
-    test_function_pass()
-    test_sequential_pass()
-    test_sequential_with_scoping()
-    test_pass_info()
-    test_print_ir()
+    pytest.main()

tests/python/relay/test_type_solver.py

@@ -16,7 +16,7 @@
 # under the License.
 import tvm
 from tvm import relay
-from nose.tools import raises
+import pytest
 
 
 def make_rel(name, args, num_inputs=None, attrs=None):
@@ -223,7 +223,8 @@ def test_backward_solving_after_child_update():
     assert solver.Resolve(t4) == tup_concrete
     assert solver.Resolve(t5) == tup_concrete
 
-@raises(tvm._ffi.base.TVMError)
+
+@pytest.mark.xfail(raises=tvm._ffi.base.TVMError)
 def test_incompatible_tuple_unification():
     solver = make_solver()
     t1 = relay.ty.IncompleteType()
@@ -238,14 +239,14 @@ def test_incompatible_tuple_unification():
     solver.Unify(tup1, tup2)
 
 
-@raises(tvm._ffi.base.TVMError)
+@pytest.mark.xfail(raises=tvm._ffi.base.TVMError)
 def test_bad_recursive_unification():
     solver = make_solver()
     t1 = relay.ty.IncompleteType()
     solver.Unify(t1, relay.ty.TupleType([t1, t1]))
 
 
-@raises(tvm._ffi.base.TVMError)
+@pytest.mark.xfail(raises=tvm._ffi.base.TVMError)
 def test_unify_invalid_global_typevars():
     solver = make_solver()
     gtv1 = relay.GlobalTypeVar('gtv1')
@@ -253,7 +254,7 @@ def test_unify_invalid_global_typevars():
     solver.Unify(gtv1, gtv2)
 
 
-@raises(tvm._ffi.base.TVMError)
+@pytest.mark.xfail(raises=tvm._ffi.base.TVMError)
 def test_incompatible_typecall_var_unification():
     solver = make_solver()
     gtv1 = relay.GlobalTypeVar('gtv1')
@@ -267,7 +268,7 @@ def test_incompatible_typecall_var_unification():
     solver.Unify(tc1, tc2)
 
 
-@raises(tvm._ffi.base.TVMError)
+@pytest.mark.xfail(raises=tvm._ffi.base.TVMError)
 def test_incompatible_typecall_args_unification():
     solver = make_solver()
     gtv = relay.GlobalTypeVar('gtv1')

tests/python/relay/test_vm.py

@@ -15,7 +15,6 @@
 # specific language governing permissions and limitations
 # under the License.
 import os
-from nose.tools import nottest, raises
 
 import tvm
 import numpy as np
@@ -23,6 +22,7 @@ from tvm import relay
 from tvm.relay.scope_builder import ScopeBuilder
 from tvm.relay.testing.config import ctx_list
 from tvm.relay.prelude import Prelude
+import pytest
 
 def check_result(args, expected_result, mod=None):
     """
@@ -328,7 +328,7 @@ def test_list_hd():
     result = veval(mod)
     tvm.testing.assert_allclose(result.asnumpy(), 3)
 
-@raises(Exception)
+@pytest.mark.xfail
 def test_list_tl_empty_list():
     mod = relay.Module()
     p = Prelude(mod)

tests/python/unittest/test_custom_datatypes_mybfloat16.py

@@ -24,7 +24,7 @@ import numpy as np
 tgt = "llvm"
 
 
-def setup():
+def setup_module():
     # You must first load the library containing the datatype implementation.
     # In this case, we have built the test functions used below right into TVM.
     # CDLL("libmybfloat16.so", RTLD_GLOBAL)
@@ -144,7 +144,7 @@ def test_bfloat_add_and_cast_FloatImm():
 
 
 if __name__ == "__main__":
-    setup()
+    setup_module()
     test_bfloat_add_and_cast_1()
     test_bfloat_add_and_cast_2()
     test_bfloat_add_and_cast_FloatImm()

tests/python/unittest/test_hybrid_script.py

@@ -14,12 +14,12 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
-import tvm, inspect, sys, traceback, numpy, nose, types, os
+import tvm, inspect, sys, traceback, numpy, pytest, types, os
 from tvm.contrib import util
 from tvm.hybrid import script
 from tvm.hybrid.runtime import HYBRID_GLOBALS
 
-@nose.tools.nottest
+@pytest.mark.skip
 def run_and_check(func, args, var_dict={}, target='llvm', sch=None, outs=None):
     def tvm_val_2_py_val(val):
         val = tvm.ir_pass.Substitute(val, var_dict)

tests/python/unittest/test_lang_schedule.py

@@ -14,7 +14,7 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
-from nose.tools import raises
+import pytest
 import tvm
 import pickle as pkl
 
@@ -129,7 +129,7 @@ def test_vectorize():
     assert s[T].iter_var_attrs[xi].iter_type == UNROLL
     assert s[T].iter_var_attrs[yi].iter_type == VECTORIZE
 
-@raises(Exception)
+@pytest.mark.xfail
 def test_vectorize_commreduce():
     V = tvm.placeholder((128,), name='V')
     ax = tvm.reduce_axis((0, 128), name='ax')

tests/python/unittest/test_pass_bound_checkers.py

@@ -14,7 +14,7 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
-from nose.tools import raises
+import pytest
 import tvm
 import numpy as np
 def collect_visit(stmt, f):
@@ -43,7 +43,7 @@ def lower(sch, args):
     stmt = tvm.ir_pass.Simplify(stmt)
     return stmt
 
-@raises(Exception)
+@pytest.mark.xfail
 def test_out_of_bounds_llvm(index_a, index_b):
     n = tvm.var("n")
     A = tvm.placeholder ((n,), name='A')
@@ -78,7 +78,7 @@ def test_in_bounds_llvm():
     c = tvm.nd.array(np.zeros(1024, dtype=C.dtype), ctx)
     fadd (a, b, c)
 
-@raises(Exception)
+@pytest.mark.xfail
 def test_out_of_bounds_vectorize_llvm(nn, index_a, index_b):
     n = tvm.convert(nn)
     a = tvm.placeholder((n), name='a')
@@ -144,7 +144,7 @@ def test_in_bounds_loop_partition_basic_llvm():
     t = tvm.nd.empty((32,), T.dtype, ctx)
     f(a, b, t)
 
-@raises(Exception)
+@pytest.mark.xfail
 def test_out_of_bounds_loop_partition_basic_llvm(index_a, index_b):
     n = tvm.var('n')
     A = tvm.placeholder((n, ), name='A')
@@ -229,7 +229,7 @@ def test_in_bounds_const_loop_partition_llvm():
         t = tvm.nd.empty((n,), T.dtype, ctx)
         f(a, b, t)
 
-@raises(Exception)
+@pytest.mark.xfail
 def test_out_of_bounds_const_loop_partition_llvm(index_a, index_b):
     with tvm.build_config(instrument_bound_checkers=True, partition_const_loop=True):
         n = 21
@@ -285,7 +285,7 @@ def test_in_bounds_conv_llvm(loop_tiling=False):
     conv_out = tvm.nd.empty ((batch_size, out_channel, out_height, out_width), tvm.float32, ctx)
     f(data_input, kernel_input, conv_out)
 
-@raises(Exception)
+@pytest.mark.xfail
 def test_out_of_bounds_conv_llvm(data_offsets, kernel_offsets, loop_tiling=False):
     HSTR = WSTR = 1
     in_channel = 128
@@ -348,7 +348,7 @@ def test_in_bounds_tensors_with_same_shapes1D_llvm():
     t = tvm.nd.empty((32,), T.dtype, ctx)
     f(a, b, t)
 
-@raises(Exception)
+@pytest.mark.xfail
 def test_out_of_bounds_tensors_with_diff_shapes1D_llvm(a_shape, b_shape, c_shape):
     n = tvm.var('n')
     k = tvm.var('k')
@@ -387,7 +387,7 @@ def test_in_bounds_tensors_with_same_shapes2D_llvm():
     t = tvm.nd.empty((32, 32), T.dtype, ctx)
     f(a, b, t)
 
-@raises(Exception)
+@pytest.mark.xfail
 def test_out_of_bounds_tensors_with_diff_shapes2D_llvm(a_shape, b_shape, c_shape):
     n = tvm.var('n')
     k = tvm.var('k')
@@ -426,7 +426,7 @@ def test_in_bounds_tensors_with_same_shapes3D_llvm():
     t = tvm.nd.empty((32, 32, 32), T.dtype, ctx)
     f(a, b, t)
 
-@raises(Exception)
+@pytest.mark.xfail
 def test_out_of_bounds_tensors_with_diff_shapes3D_llvm(a_shape, b_shape, c_shape):
     n = tvm.var('n')
     k = tvm.var('k')
@@ -446,7 +446,7 @@ def test_out_of_bounds_tensors_with_diff_shapes3D_llvm(a_shape, b_shape, c_shape
     t = tvm.nd.empty((c_shape[0],c_shape[1],c_shape[2]), T.dtype, ctx)
     f(a, b, t)
 
-@raises(Exception)
+@pytest.mark.xfail
 def test_out_of_bounds_tensors_with_zero_shape_op_with_not_zero_shape_llvm():
     if not tvm.module.enabled("llvm"):
         return

tests/python/unittest/test_pass_split_host_device.py

@@ -14,10 +14,10 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
-from nose.tools import raises
+import pytest
 import tvm
 
-@raises(Exception)
+@pytest.mark.xfail
 def test_loop_dependent_allocate():
     N = tvm.var("N")
     A = tvm.placeholder((2*N,), "float32", "A")

tests/python/unittest/test_runtime_micro.py

@@ -17,7 +17,6 @@
 
 import os
 
-from nose.tools import nottest
 import numpy as np
 import tvm
 from tvm.contrib import graph_runtime, util

tests/python/unittest/test_runtime_vm_profiler.py

@@ -18,11 +18,11 @@ import os
 import tvm
 import numpy as np
 
-from nose.tools import nottest
+import pytest
 from tvm import relay
 from tvm.relay.testing import resnet
 
-@nottest
+@pytest.mark.skip
 def test_basic():
     mod, params = resnet.get_workload()
     compiler = relay.profiler_vm.VMCompilerProfiler()

tests/scripts/task_python_frontend.sh

@@ -27,28 +27,28 @@ export OMP_NUM_THREADS=1
 make cython3
 
 echo "Running relay TFLite frontend test..."
-python3 -m nose -v tests/python/frontend/tflite
+python3 -m pytest -v tests/python/frontend/tflite
 
 echo "Running relay MXNet frontend test..."
-python3 -m nose -v tests/python/frontend/mxnet
+python3 -m pytest -v tests/python/frontend/mxnet
 
 echo "Running relay Keras frontend test..."
-python3 -m nose -v tests/python/frontend/keras
+python3 -m pytest -v tests/python/frontend/keras
 
 echo "Running relay ONNX frontend test..."
-python3 -m nose -v tests/python/frontend/onnx
+python3 -m pytest -v tests/python/frontend/onnx
 
 echo "Running relay CoreML frontend test..."
-python3 -m nose -v tests/python/frontend/coreml
+python3 -m pytest -v tests/python/frontend/coreml
 
 echo "Running nnvm to relay frontend test..."
-python3 -m nose -v tests/python/frontend/nnvm_to_relay
+python3 -m pytest -v tests/python/frontend/nnvm_to_relay
 
 echo "Running relay Tensorflow frontend test..."
-python3 -m nose -v tests/python/frontend/tensorflow
+python3 -m pytest -v tests/python/frontend/tensorflow
 
 echo "Running relay caffe2 frontend test..."
-python3 -m nose -v tests/python/frontend/caffe2
+python3 -m pytest -v tests/python/frontend/caffe2
 
 echo "Running relay DarkNet frontend test..."
-python3 -m nose -v tests/python/frontend/darknet
+python3 -m pytest -v tests/python/frontend/darknet

tests/scripts/task_python_integration.sh

@@ -33,13 +33,13 @@ rm -rf lib
 make
 cd ../..
 
-python3 -m nose -v apps/extension/tests
+python3 -m pytest -v apps/extension/tests
 
-TVM_FFI=ctypes python3 -m nose -v tests/python/integration
-TVM_FFI=ctypes python3 -m nose -v tests/python/contrib
+TVM_FFI=ctypes python3 -m pytest -v tests/python/integration
+TVM_FFI=ctypes python3 -m pytest -v tests/python/contrib
 
-TVM_FFI=ctypes python3 -m nose -v tests/python/relay
+TVM_FFI=ctypes python3 -m pytest -v tests/python/relay
 
 # Do not enable OpenGL
-# TVM_FFI=cython python -m nose -v tests/webgl
-# TVM_FFI=ctypes python3 -m nose -v tests/webgl
+# TVM_FFI=cython python -m pytest -v tests/webgl
+# TVM_FFI=ctypes python3 -m pytest -v tests/webgl

tests/scripts/task_python_legacy.sh

@@ -27,29 +27,29 @@ export OMP_NUM_THREADS=1
 make cython3
 
 echo "Running nnvm unittest..."
-python3 -m nose -v nnvm/tests/python/unittest
+python3 -m pytest -v nnvm/tests/python/unittest
 
 
 echo "Running nnvm compiler test..."
-python3 -m nose -v nnvm/tests/python/compiler
+python3 -m pytest -v nnvm/tests/python/compiler
 
 echo "Running nnvm ONNX frontend test..."
-python3 -m nose -v nnvm/tests/python/frontend/onnx
+python3 -m pytest -v nnvm/tests/python/frontend/onnx
 
 echo "Running nnvm MXNet frontend test..."
-python3 -m nose -v nnvm/tests/python/frontend/mxnet
+python3 -m pytest -v nnvm/tests/python/frontend/mxnet
 
 echo "Running nnvm DarkNet frontend test..."
-python3 -m nose -v nnvm/tests/python/frontend/darknet
+python3 -m pytest -v nnvm/tests/python/frontend/darknet
 
 echo "Running nnvm Keras frontend test..."
-python3 -m nose -v nnvm/tests/python/frontend/keras
+python3 -m pytest -v nnvm/tests/python/frontend/keras
 
 echo "Running nnvm Tensorflow frontend test..."
-python3 -m nose -v nnvm/tests/python/frontend/tensorflow
+python3 -m pytest -v nnvm/tests/python/frontend/tensorflow
 
 echo "Running nnvm CoreML frontend test..."
-python3 -m nose -v nnvm/tests/python/frontend/coreml
+python3 -m pytest -v nnvm/tests/python/frontend/coreml
 
 echo "Running nnvm Caffe2 frontend test..."
-python3 -m nose -v nnvm/tests/python/frontend/caffe2
+python3 -m pytest -v nnvm/tests/python/frontend/caffe2

tests/scripts/task_python_nightly.sh

@@ -25,6 +25,6 @@ export PYTHONPATH=python:topi/python
 make cython3
 
 rm -rf python/tvm/*.pyc python/tvm/*/*.pyc python/tvm/*/*/*.pyc
-rm -rf topi/python/topi/*.pyc topi/python/topi/*/*.pyc topi/python/topi/*/*/*.pyc topi/python/topi/*/*/*/*.pyc 
+rm -rf topi/python/topi/*.pyc topi/python/topi/*/*.pyc topi/python/topi/*/*/*.pyc topi/python/topi/*/*/*/*.pyc
 
-python3 -m nose -v topi/tests/python/nightly
+python3 -m pytest -v topi/tests/python/nightly

tests/scripts/task_python_topi.sh

@@ -25,6 +25,6 @@ export PYTHONPATH=python:topi/python
 make cython3
 
 rm -rf python/tvm/*.pyc python/tvm/*/*.pyc python/tvm/*/*/*.pyc
-rm -rf topi/python/topi/*.pyc topi/python/topi/*/*.pyc topi/python/topi/*/*/*.pyc topi/python/topi/*/*/*/*.pyc 
+rm -rf topi/python/topi/*.pyc topi/python/topi/*/*.pyc topi/python/topi/*/*/*.pyc topi/python/topi/*/*/*/*.pyc
 
-python3 -m nose -v topi/tests/python
+python3 -m pytest -v topi/tests/python

tests/scripts/task_python_unittest.sh

@@ -23,6 +23,6 @@ export PYTHONPATH=python:topi/python
 
 rm -rf python/tvm/*.pyc python/tvm/*/*.pyc python/tvm/*/*/*.pyc
 
-TVM_FFI=ctypes python3 -m nose -v tests/python/unittest
+TVM_FFI=ctypes python3 -m pytest -v tests/python/unittest
 make cython3
-TVM_FFI=cython python3 -m nose -v tests/python/unittest
+TVM_FFI=cython python3 -m pytest -v tests/python/unittest

tests/scripts/task_python_vta.sh

@@ -29,11 +29,11 @@ make cython3
 
 # Run unit tests in functional/fast simulator
 echo "Running unittest in fsim..."
-python3 -m nose -v vta/tests/python/unittest
+python3 -m pytest -v vta/tests/python/unittest
 
 # Run unit tests in functional/fast simulator
 echo "Running integration test in fsim..."
-python3 -m nose -v vta/tests/python/integration
+python3 -m pytest -v vta/tests/python/integration
 
 # # Build VTA chisel design and verilator simulator
 # (make -C vta/hardware/chisel/)
@@ -43,11 +43,11 @@ python3 -m nose -v vta/tests/python/integration
 
 # # Run unit tests in cycle accurate simulator
 # echo "Running unittest in tsim..."
-# python3 -m nose -v vta/tests/python/unittest
+# python3 -m pytest -v vta/tests/python/unittest
 
 # # Run unit tests in cycle accurate simulator
 # echo "Running integration test in tsim..."
-# python3 -m nose -v vta/tests/python/integration
+# python3 -m pytest -v vta/tests/python/integration
 
 # # Reset default fsim simulation
-# cp vta/config/fsim_sample.json vta/config/vta_config.json
+# cp vta/config/fsim_sample.json vta/config/vta_config.json
\ No newline at end of file

tests/scripts/task_verilog_test.sh

@@ -21,5 +21,5 @@ set -u
 
 export PYTHONPATH=python
 make verilog
-nosetests -v tests/verilog/unittest
-nosetests -v tests/verilog/integration
+python3 -m pytest -v tests/verilog/unittest
+python3 -m pytest -v tests/verilog/integration

tests/travis/run_test.sh

@@ -6,9 +6,9 @@
 # to you under the Apache License, Version 2.0 (the
 # "License"); you may not use this file except in compliance
 # with the License.  You may obtain a copy of the License at
-# 
+#
 #   http://www.apache.org/licenses/LICENSE-2.0
-# 
+#
 # Unless required by applicable law or agreed to in writing,
 # software distributed under the License is distributed on an
 # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
@@ -55,7 +55,6 @@ if [ ${TASK} == "python_test" ] || [ ${TASK} == "all_test" ]; then
     if [ ${TRAVIS_OS_NAME} == "osx" ]; then
         ./tests/scripts/task_python_unittest.sh || exit -1
     else
-        nosetests -v tests/python/unittest || exit -1
-        nosetests3 -v tests/python/unittest || exit -1
+        pytests3 -v tests/python/unittest || exit -1
     fi
 fi

tests/travis/setup.sh

@@ -6,9 +6,9 @@
 # to you under the Apache License, Version 2.0 (the
 # "License"); you may not use this file except in compliance
 # with the License.  You may obtain a copy of the License at
-# 
+#
 #   http://www.apache.org/licenses/LICENSE-2.0
-# 
+#
 # Unless required by applicable law or agreed to in writing,
 # software distributed under the License is distributed on an
 # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
@@ -20,8 +20,7 @@ if [ ${TASK} == "python_test" ] || [ ${TASK} == "all_test" ]; then
     if [ ${TRAVIS_OS_NAME} == "osx" ]; then
         brew update
         brew install python3
-        python -m pip install --user nose numpy cython
-        python3 -m pip install --user nose numpy cython
+        python3 -m pip install --user pytest numpy cython
     fi
 fi
 

topi/tests/python/test_topi_group_conv2d_NCHWc_int8.py

@@ -24,7 +24,7 @@ import topi
 import topi.testing
 from tvm.contrib.pickle_memoize import memoize
 from topi.util import get_const_tuple
-from nose.tools import nottest
+import pytest
 
 from common import get_all_backend
 
@@ -103,7 +103,7 @@ def verify_group_conv2d_NCHWc_int8(batch, in_channel, groups, in_size, num_filte
         with autotvm.tophub.context(device):  # load tophub pre-tuned parameters
             check_device(device)
 
-@nottest
+@pytest.mark.skip
 def test_conv2d_NCHWc():
     # ResNet50 workloads
     verify_group_conv2d_NCHWc_int8(1, 256, 32, 224, 64, 7, 2, 3)