fix test data

kendryte · Jul 24, 2023 · 655b261 · 655b261
1 parent 5b5b958
commit 655b261
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/src/Nncase.Tests/Rules/Neutral/UnitTestReshapeMatMul.cs b/src/Nncase.Tests/Rules/Neutral/UnitTestReshapeMatMul.cs
@@ -39,11 +39,12 @@ public void TestTo3D(int[] shapeA, int[] shapeB)
         TestMatched<ReshapeMatMul>(mm);
     }
 
-    [Fact]
-    public void TestNop()
+    [Theory]
+    [MemberData(nameof(NopMatMulShapeData))]
+    public void TestNop(int[] shapeA, int[] shapeB)
     {
-        var lhs = Testing.Rand<float>(1, 3, 24, 24);
-        var rhs = Testing.Rand<float>(3, 24, 24);
+        var lhs = Testing.Rand<float>(shapeA);
+        var rhs = Testing.Rand<float>(shapeB);
         var mm = MatMul(lhs, rhs);
         TestNotMatch<ReshapeMatMul>(mm);
     }