adjust tests

Author: Sheng Zha

tests/python/unittest/test_contrib_control_flow.py

@@ -27,6 +27,7 @@
 from mxnet.attribute import AttrScope
 from common import with_seed
 
+mx.reset_np()
 
 @with_seed()
 def test_while_loop_simple_forward():

tests/python/unittest/test_contrib_io.py

@@ -15,12 +15,15 @@
 # specific language governing permissions and limitations
 # under the License.
 
+import mxnet as mx
 import mxnet.ndarray as nd
 from mxnet.gluon.data.vision.datasets import *
 from mxnet.gluon.data.dataloader import *
 from mxnet.contrib.io import *
 from mxnet.test_utils import *
 
+mx.reset_np()
+
 def test_contrib_DataLoaderIter():
     def test_mnist_batches(batch_size, expected, last_batch='discard'):
         dataset = MNIST(train=False)

tests/python/unittest/test_contrib_stes_op.py

@@ -20,6 +20,7 @@
 from mxnet import nd, autograd, gluon
 from mxnet.test_utils import default_context
 
+mx.reset_np()
 
 class RoundSTENET(gluon.HybridBlock):
     def __init__(self, w_init, **kwargs):

tests/python/unittest/test_contrib_text.py

@@ -20,12 +20,14 @@
 from collections import Counter
 
 from common import assertRaises
+import mxnet as mx
 from mxnet import ndarray as nd
 from mxnet.test_utils import *
 from mxnet.contrib import text
 
 import pytest
 
+mx.reset_np()
 
 def _get_test_str_of_tokens(token_delim, seq_delim):
     seq1 = token_delim + token_delim.join(['Life', 'is', 'great', '!']) + token_delim + seq_delim

tests/python/unittest/test_exc_handling.py

@@ -25,6 +25,7 @@
 from mxnet.test_utils import assert_exception, default_context, set_default_context, use_np
 import pytest
 
+mx.reset_np()
 
 @with_seed()
 @pytest.mark.skipif(os.environ.get('MXNET_ENGINE_TYPE') == 'NaiveEngine',

tests/python/unittest/test_gluon.py

@@ -38,6 +38,8 @@
 import random
 import tempfile
 
+mx.reset_np()
+
 @with_seed()
 def test_parameter():
     p = gluon.Parameter('weight', shape=(10, 10))
@@ -192,11 +194,11 @@ def __init__(self, **kwargs):
 
     net = Net()
     lines = str(net.collect_params()).splitlines()
-    
+
     assert 'dense0.weight' in lines[0]
     assert '(10, 5)' in lines[0]
     assert 'float32' in lines[0]
-    
+
 
 @with_seed()
 def test_collect_parameters():
@@ -1735,7 +1737,7 @@ def mon_callback(node_name, opr_name, arr):
     model.add(mx.gluon.nn.AvgPool1D())
     model.initialize()
     model.hybridize()
-    check_name(model, [model[0].name + '_fwd_data', model[0].name + '_fwd_output'], 
+    check_name(model, [model[0].name + '_fwd_data', model[0].name + '_fwd_output'],
                expected_opr_names=["Pooling"], monitor_all=True)
 
     # stack two layers and test

tests/python/unittest/test_gluon_batch_processor.py

@@ -29,6 +29,8 @@
 from mxnet.gluon.contrib.estimator.batch_processor import BatchProcessor
 import pytest
 
+mx.reset_np()
+
 def _get_test_network():
     net = nn.Sequential()
     net.add(nn.Dense(4, activation='relu', flatten=False))

tests/python/unittest/test_gluon_contrib.py

@@ -28,6 +28,7 @@
 from common import setup_module, with_seed, teardown_module
 import numpy as np
 
+mx.reset_np()
 
 def check_rnn_cell(cell, in_shape=(10, 50), out_shape=(10, 100), begin_state=None):
     inputs = [mx.sym.Variable('rnn_t%d_data'%i) for i in range(3)]

tests/python/unittest/test_gluon_data_vision.py

@@ -30,6 +30,8 @@
 
 import numpy as np
 
+mx.reset_np()
+
 @with_seed()
 def test_to_tensor():
     # 3D Input

tests/python/unittest/test_gluon_estimator.py

@@ -28,6 +28,7 @@
 from mxnet.gluon.contrib.estimator import *
 from mxnet.gluon.contrib.estimator.event_handler import *
 
+mx.reset_np()
 
 def _get_test_network(params=None):
     net = nn.Sequential()