fix: tests

Author: NorbertKlockiewicz

apps/computer-vision/app/style_transfer/index.tsx

@@ -44,7 +44,6 @@ export default function StyleTransferScreen() {
         const output = await model.forward(imageUri);
         const height = output.sizes[0];
         const width = output.sizes[1];
-        // Native already returns RGBA uint8 — use directly
         const skData = Skia.Data.fromBytes(output.dataPtr);
         const img = Skia.Image.MakeImage(
           {

packages/react-native-executorch/common/rnexecutorch/tests/CMakeLists.txt

@@ -148,8 +148,11 @@ add_rn_test(BaseModelTests integration/BaseModelTest.cpp)
 add_rn_test(ClassificationTests integration/ClassificationTest.cpp
     SOURCES
         ${RNEXECUTORCH_DIR}/models/classification/Classification.cpp
+        ${RNEXECUTORCH_DIR}/models/VisionModel.cpp
+        ${RNEXECUTORCH_DIR}/utils/FrameProcessor.cpp
+        ${RNEXECUTORCH_DIR}/utils/FrameExtractor.cpp
         ${IMAGE_UTILS_SOURCES}
-    LIBS opencv_deps
+    LIBS opencv_deps android
 )
 
 add_rn_test(ObjectDetectionTests integration/ObjectDetectionTest.cpp
@@ -167,8 +170,11 @@ add_rn_test(ImageEmbeddingsTests integration/ImageEmbeddingsTest.cpp
     SOURCES
         ${RNEXECUTORCH_DIR}/models/embeddings/image/ImageEmbeddings.cpp
         ${RNEXECUTORCH_DIR}/models/embeddings/BaseEmbeddings.cpp
+        ${RNEXECUTORCH_DIR}/models/VisionModel.cpp
+        ${RNEXECUTORCH_DIR}/utils/FrameProcessor.cpp
+        ${RNEXECUTORCH_DIR}/utils/FrameExtractor.cpp
         ${IMAGE_UTILS_SOURCES}
-    LIBS opencv_deps
+    LIBS opencv_deps android
 )
 
 add_rn_test(TextEmbeddingsTests integration/TextEmbeddingsTest.cpp
@@ -182,8 +188,11 @@ add_rn_test(TextEmbeddingsTests integration/TextEmbeddingsTest.cpp
 add_rn_test(StyleTransferTests integration/StyleTransferTest.cpp
     SOURCES
         ${RNEXECUTORCH_DIR}/models/style_transfer/StyleTransfer.cpp
+        ${RNEXECUTORCH_DIR}/models/VisionModel.cpp
+        ${RNEXECUTORCH_DIR}/utils/FrameProcessor.cpp
+        ${RNEXECUTORCH_DIR}/utils/FrameExtractor.cpp
         ${IMAGE_UTILS_SOURCES}
-    LIBS opencv_deps
+    LIBS opencv_deps android
 )
 
 add_rn_test(VADTests integration/VoiceActivityDetectionTest.cpp
@@ -244,8 +253,10 @@ add_rn_test(OCRTests integration/OCRTest.cpp
         ${RNEXECUTORCH_DIR}/models/ocr/utils/DetectorUtils.cpp
         ${RNEXECUTORCH_DIR}/models/ocr/utils/RecognitionHandlerUtils.cpp
         ${RNEXECUTORCH_DIR}/models/ocr/utils/RecognizerUtils.cpp
+        ${RNEXECUTORCH_DIR}/utils/FrameProcessor.cpp
+        ${RNEXECUTORCH_DIR}/utils/FrameExtractor.cpp
         ${IMAGE_UTILS_SOURCES}
-    LIBS opencv_deps
+    LIBS opencv_deps android
 )
 
 add_rn_test(VerticalOCRTests integration/VerticalOCRTest.cpp
@@ -258,6 +269,8 @@ add_rn_test(VerticalOCRTests integration/VerticalOCRTest.cpp
         ${RNEXECUTORCH_DIR}/models/ocr/utils/DetectorUtils.cpp
         ${RNEXECUTORCH_DIR}/models/ocr/utils/RecognitionHandlerUtils.cpp
         ${RNEXECUTORCH_DIR}/models/ocr/utils/RecognizerUtils.cpp
+        ${RNEXECUTORCH_DIR}/utils/FrameProcessor.cpp
+        ${RNEXECUTORCH_DIR}/utils/FrameExtractor.cpp
         ${IMAGE_UTILS_SOURCES}
-    LIBS opencv_deps
+    LIBS opencv_deps android
 )

packages/react-native-executorch/common/rnexecutorch/tests/integration/StyleTransferTest.cpp

@@ -60,16 +60,18 @@ TEST(StyleTransferGenerateTests, MalformedURIThrows) {
 TEST(StyleTransferGenerateTests, ValidImageReturnsNonNull) {
   StyleTransfer model(kValidStyleTransferModelPath, nullptr);
   auto result = model.generateFromString(kValidTestImagePath);
-  EXPECT_NE(result, nullptr);
+  EXPECT_NE(result.dataPtr, nullptr);
+  EXPECT_GT(result.width, 0);
+  EXPECT_GT(result.height, 0);
 }
 
 TEST(StyleTransferGenerateTests, MultipleGeneratesWork) {
   StyleTransfer model(kValidStyleTransferModelPath, nullptr);
   EXPECT_NO_THROW((void)model.generateFromString(kValidTestImagePath));
   auto result1 = model.generateFromString(kValidTestImagePath);
   auto result2 = model.generateFromString(kValidTestImagePath);
-  EXPECT_NE(result1, nullptr);
-  EXPECT_NE(result2, nullptr);
+  EXPECT_NE(result1.dataPtr, nullptr);
+  EXPECT_NE(result2.dataPtr, nullptr);
 }
 
 TEST(StyleTransferInheritedTests, GetInputShapeWorks) {