Create MI tests for Tensors: ARGMAX, MAX_POOL2D

Signed-off-by: James Ward <james.ward@arm.com>
Change-Id: I3682311cbdffa52fa88ae1b5779c615e2f062807

3 files changed, 128 insertions, 0 deletions

diff --git a/operators/tensor/argmax/argmax_ERRORIF_WrongOutputList_10x3_f32_axis0/desc.json b/operators/tensor/argmax/argmax_ERRORIF_WrongOutputList_10x3_f32_axis0/desc.json
new file mode 100644
index 000000000..671c6cfc2
--- /dev/null
+++ b/operators/tensor/argmax/argmax_ERRORIF_WrongOutputList_10x3_f32_axis0/desc.json
@@ -0,0 +1,21 @@
+{
+  "tosa_file": "test.json",
+  "ifm_name": [
+    "input-0"
+  ],
+  "ifm_file": [
+    "input-0.npy"
+  ],
+  "ofm_name": [
+    "result-0"
+  ],
+  "ofm_file": [
+    "result-0.npy"
+  ],
+  "expected_return_code": 2,
+  "expected_failure": true,
+  "expected_failure_desc": "Op output list does not match expected output",
+  "profile": [
+    "tosa-mi"
+  ]
+}
\ No newline at end of file
diff --git a/operators/tensor/argmax/argmax_ERRORIF_WrongOutputList_10x3_f32_axis0/input-0.json b/operators/tensor/argmax/argmax_ERRORIF_WrongOutputList_10x3_f32_axis0/input-0.json
new file mode 100644
index 000000000..fcbaed64f
--- /dev/null
+++ b/operators/tensor/argmax/argmax_ERRORIF_WrongOutputList_10x3_f32_axis0/input-0.json
@@ -0,0 +1,55 @@
+{
+  "type": "float32",
+  "data": [
+    [
+      0.11928489804267883,
+      0.005696401931345463,
+      0.38493648171424866
+    ],
+    [
+      0.24898239970207214,
+      0.9030187129974365,
+      0.6030975580215454
+    ],
+    [
+      0.6353207230567932,
+      0.7337390780448914,
+      0.6623497009277344
+    ],
+    [
+      0.08567645400762558,
+      0.5335094332695007,
+      0.4929577708244324
+    ],
+    [
+      0.41566771268844604,
+      0.8635202050209045,
+      0.016220033168792725
+    ],
+    [
+      0.7899508476257324,
+      0.5882290601730347,
+      0.9025305509567261
+    ],
+    [
+      0.9789959788322449,
+      0.4726513922214508,
+      0.15101556479930878
+    ],
+    [
+      0.9990746974945068,
+      0.6687710881233215,
+      0.3285938799381256
+    ],
+    [
+      0.06079389154911041,
+      0.8828278183937073,
+      0.21333321928977966
+    ],
+    [
+      0.9132226705551147,
+      0.506874144077301,
+      0.45885956287384033
+    ]
+  ]
+}
\ No newline at end of file
diff --git a/operators/tensor/argmax/argmax_ERRORIF_WrongOutputList_10x3_f32_axis0/test.json b/operators/tensor/argmax/argmax_ERRORIF_WrongOutputList_10x3_f32_axis0/test.json
new file mode 100644
index 000000000..f59861a84
--- /dev/null
+++ b/operators/tensor/argmax/argmax_ERRORIF_WrongOutputList_10x3_f32_axis0/test.json
@@ -0,0 +1,52 @@
+{
+  version: {
+    _major: 0,
+    _minor: 41,
+    _patch: 0,
+    _draft: true
+  },
+  blocks: [
+    {
+      name: "main",
+      operators: [
+        {
+          op: "ARGMAX",
+          attribute_type: "AxisAttribute",
+          attribute: {
+            axis: 0
+          },
+          inputs: [
+            "input-0"
+          ],
+          outputs: [
+            "result-0",
+            "eiDummyOutput"
+          ]
+        }
+      ],
+      tensors: [
+        {
+          name: "input-0",
+          shape: [
+            10,
+            3
+          ],
+          type: "FP32"
+        },
+        {
+          name: "result-0",
+          shape: [
+            3
+          ],
+          type: "INT32"
+        }
+      ],
+      inputs: [
+        "input-0"
+      ],
+      outputs: [
+        "result-0"
+      ]
+    }
+  ]
+}