From c215a4ca625901f1c89ad21668ce3bb46fa3ea10 Mon Sep 17 00:00:00 2001
From: NAUD Maxence <maxence.naud@cea.fr>
Date: Tue, 17 Oct 2023 12:26:25 +0000
Subject: [PATCH] [Upd] Slice test
---
unit_tests/operator/Test_SliceImpl.cpp | 16 ++++++++++++----
1 file changed, 12 insertions(+), 4 deletions(-)
diff --git a/unit_tests/operator/Test_SliceImpl.cpp b/unit_tests/operator/Test_SliceImpl.cpp
index 0bf12f9b..bc129dae 100644
--- a/unit_tests/operator/Test_SliceImpl.cpp
+++ b/unit_tests/operator/Test_SliceImpl.cpp
@@ -33,8 +33,10 @@ TEST_CASE("[cpu/operator] Slice(forward)", "[Slice]") {
mySlice->getOperator()->associateInput(0,input0);
mySlice->getOperator()->computeOutputDims();
mySlice->forward();
- mySlice->getOperator()->output(0).print();
+ // mySlice->getOperator()->output(0).print();
REQUIRE(mySlice->getOperator()->output(0) == *expectedOutput);
+ REQUIRE(mySlice->getOperator()->output(0).dims() == expectedOutput->dims());
+ REQUIRE(mySlice->getOperator()->output(0).dataType() == expectedOutput->dataType());
}
SECTION("2D Tensor") {
@@ -57,8 +59,10 @@ TEST_CASE("[cpu/operator] Slice(forward)", "[Slice]") {
mySlice->getOperator()->associateInput(0,input0);
mySlice->getOperator()->computeOutputDims();
mySlice->forward();
- mySlice->getOperator()->output(0).print();
+ // mySlice->getOperator()->output(0).print();
REQUIRE(*mySlice->getOperator()->getOutput(0) == *expectedOutput);
+ REQUIRE(mySlice->getOperator()->output(0).dims() == expectedOutput->dims());
+ REQUIRE(mySlice->getOperator()->output(0).dataType() == expectedOutput->dataType());
}
SECTION("3D Tensor") {
@@ -88,8 +92,10 @@ TEST_CASE("[cpu/operator] Slice(forward)", "[Slice]") {
mySlice->getOperator()->associateInput(0,input0);
mySlice->getOperator()->computeOutputDims();
mySlice->forward();
- mySlice->getOperator()->output(0).print();
+ // mySlice->getOperator()->output(0).print();
REQUIRE(mySlice->getOperator()->output(0) == *expectedOutput);
+ REQUIRE(mySlice->getOperator()->output(0).dims() == expectedOutput->dims());
+ REQUIRE(mySlice->getOperator()->output(0).dataType() == expectedOutput->dataType());
}
SECTION("4D Tensor") {
@@ -148,7 +154,9 @@ TEST_CASE("[cpu/operator] Slice(forward)", "[Slice]") {
mySlice->getOperator()->associateInput(0,input0);
mySlice->getOperator()->computeOutputDims();
mySlice->forward();
- mySlice->getOperator()->output(0).print();
+ // mySlice->getOperator()->output(0).print();
REQUIRE(mySlice->getOperator()->output(0) == *expectedOutput);
+ REQUIRE(mySlice->getOperator()->output(0).dims() == expectedOutput->dims());
+ REQUIRE(mySlice->getOperator()->output(0).dataType() == expectedOutput->dataType());
}
}
\ No newline at end of file
--
GitLab