[test] test_nbla_reshape with -1 in 0-th axis

sony · Aug 8, 2018 · 79dc7d9 · 79dc7d9
1 parent c308fa2
commit 79dc7d9
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/python/test/cpp/test_nbla.py b/python/test/cpp/test_nbla.py
@@ -84,7 +84,7 @@ def check_nbla_infer(tmpdir, x, y, batch_size):
 
 
 @pytest.mark.parametrize('batch_size', [1, 4])
-@pytest.mark.parametrize("shape", [(10, 56, -1), (10, 56, 7, 20, 10)])
+@pytest.mark.parametrize("shape", [(10, 56, -1), (-1, 56, 7, 20, 10)])
 def test_nbla_reshape(tmpdir, batch_size, shape):
 
     x = nn.Variable([10, 1, 28, 28, 10, 10])

diff --git a/python/test/test_nnp_graph.py b/python/test/test_nnp_graph.py
@@ -98,7 +98,7 @@ def check_nnp_graph_save_load(tmpdir, x, y, batch_size, variable_batch_size):
 
 @pytest.mark.parametrize('variable_batch_size', [False, True])
 @pytest.mark.parametrize('batch_size', [1, 4])
-@pytest.mark.parametrize("shape", [(10, 56, -1), (10, 56, 7, 20, 10)])
+@pytest.mark.parametrize("shape", [(10, 56, -1), (-1, 56, 7, 20, 10)])
 def test_nnp_graph_reshape(tmpdir, variable_batch_size, batch_size, shape):
     x = nn.Variable([10, 1, 28, 28, 10, 10])
     y = F.reshape(x, shape=shape)

diff --git a/python/test/utils/test_save_load.py b/python/test/utils/test_save_load.py
@@ -70,7 +70,7 @@ def check_save_load(tmpdir, x, y, variable_batch_size):
 
 
 @pytest.mark.parametrize("variable_batch_size", [False, True])
-@pytest.mark.parametrize("shape", [(10, 56, -1), (10, 56, 7, 20, 10)])
+@pytest.mark.parametrize("shape", [(10, 56, -1), (-1, 56, 7, 20, 10)])
 def test_save_load_reshape(tmpdir, variable_batch_size, shape):
     x = nn.Variable([10, 1, 28, 28, 10, 10])
     y = F.reshape(x, shape=shape)