1 more addition to test

apache · Apr 17, 2020 · 9129ba2 · 9129ba2
1 parent de6125e
commit 9129ba2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/python/unittest/test_operator.py b/tests/python/unittest/test_operator.py
@@ -9961,7 +9961,7 @@ def test_elementwise_ops_on_misaligned_input():
 @with_seed()
 def test_broadcast_ops_on_misaligned_input():
     dtypes = ['float16', 'float32', 'float64']
-    lead_dims = [2,4,6,10]
+    lead_dims = [2,3,4,6,10]
 
     for dtype in dtypes:
         for lead_dim in lead_dims:
@@ -9990,7 +9990,7 @@ def test_broadcast_ops_on_misaligned_input():
 @with_seed()
 def test_broadcast_ops_on_misaligned_input_oneside():
     dtypes = ['float16', 'float32', 'float64']
-    lead_dims = [2,4,6,10]
+    lead_dims = [2,3,4,6,10]
 
     for dtype in dtypes:
         for lead_dim in lead_dims: