react-native-executorch 0.7.0 → 0.7.2

package/common/rnexecutorch/tests/CMakeLists.txt

@@ -1,253 +0,0 @@
-if(NOT ANDROID_ABI)
-    message(FATAL_ERROR "Tests can be only built for Android simulator")
-endif()
-
-cmake_minimum_required(VERSION 3.13)
-project(RNExecutorchTests)
-
-set(CMAKE_CXX_STANDARD 20)
-set(CMAKE_CXX_STANDARD_REQUIRED TRUE)
-
-# tests/                         <- CMAKE_SOURCE_DIR (this file's location)
-# rnexecutorch/                  <- RNEXECUTORCH_DIR (parent of tests)
-# common/                        <- COMMON_DIR
-# react-native-executorch/       <- PACKAGE_ROOT
-# <monorepo-root>/               <- MONOREPO_ROOT
-# <monorepo-root>/third-party/   <- THIRD_PARTY_DIR
-set(RNEXECUTORCH_DIR "${CMAKE_SOURCE_DIR}/..")
-set(COMMON_DIR "${RNEXECUTORCH_DIR}/..")
-set(PACKAGE_ROOT "${COMMON_DIR}/..")
-set(MONOREPO_ROOT "${PACKAGE_ROOT}/../..")
-set(THIRD_PARTY_DIR "${MONOREPO_ROOT}/third-party")
-set(REACT_NATIVE_DIR "${MONOREPO_ROOT}/node_modules/react-native")
-set(ANDROID_THIRD_PARTY "${PACKAGE_ROOT}/third-party/android/libs/")
-set(TOKENIZERS_DIR "${PACKAGE_ROOT}/third-party/include/executorch/extension/llm/tokenizers/include")
-
-# Add Gtest as a subdirectory
-add_subdirectory(${THIRD_PARTY_DIR}/googletest ${PROJECT_BINARY_DIR}/googletest)
-
-# ExecuTorch Prebuilt binaries
-add_library(executorch_prebuilt SHARED IMPORTED)
-set_target_properties(executorch_prebuilt PROPERTIES
-    IMPORTED_LOCATION "${ANDROID_THIRD_PARTY}/executorch/${ANDROID_ABI}/libexecutorch.so"
-)
-
-# pthreadpool and cpuinfo (needed for OpenMP/OpenCV)
-if(ANDROID_ABI STREQUAL "arm64-v8a")
-  add_library(pthreadpool SHARED IMPORTED)
-  set_target_properties(pthreadpool PROPERTIES
-      IMPORTED_LOCATION "${ANDROID_THIRD_PARTY}/pthreadpool/${ANDROID_ABI}/libpthreadpool.so"
-  )
-
-  add_library(cpuinfo SHARED IMPORTED)
-  set_target_properties(cpuinfo PROPERTIES
-      IMPORTED_LOCATION "${ANDROID_THIRD_PARTY}/cpuinfo/${ANDROID_ABI}/libcpuinfo.so"
-  )
-
-  set(EXECUTORCH_LIBS pthreadpool cpuinfo)
-else()
-  set(EXECUTORCH_LIBS "")
-endif()
-
-# OpenCV (Interface Library)
-set(OPENCV_LIBS_DIR "${ANDROID_THIRD_PARTY}/opencv/${ANDROID_ABI}")
-set(OPENCV_THIRD_PARTY_DIR "${ANDROID_THIRD_PARTY}/opencv-third-party/${ANDROID_ABI}")
-
-if(ANDROID_ABI STREQUAL "arm64-v8a")
-  set(OPENCV_THIRD_PARTY_LIBS
-    "${OPENCV_THIRD_PARTY_DIR}/libkleidicv_hal.a"
-    "${OPENCV_THIRD_PARTY_DIR}/libkleidicv_thread.a"
-    "${OPENCV_THIRD_PARTY_DIR}/libkleidicv.a"
-  )
-elseif(ANDROID_ABI STREQUAL "x86_64")
-  set(OPENCV_THIRD_PARTY_LIBS "")
-endif()
-
-
-add_library(opencv_deps INTERFACE)
-target_link_libraries(opencv_deps INTERFACE
-    ${OPENCV_LIBS_DIR}/libopencv_core.a
-    ${OPENCV_LIBS_DIR}/libopencv_features2d.a
-    ${OPENCV_LIBS_DIR}/libopencv_highgui.a
-    ${OPENCV_LIBS_DIR}/libopencv_imgproc.a
-    ${OPENCV_LIBS_DIR}/libopencv_photo.a
-    ${OPENCV_LIBS_DIR}/libopencv_video.a
-    ${OPENCV_THIRD_PARTY_LIBS}
-    ${EXECUTORCH_LIBS}
-    z
-    dl
-    m
-    log
-)
-target_link_options(opencv_deps INTERFACE -fopenmp -static-openmp)
-
-add_library(tokenizers_deps INTERFACE)
-target_include_directories(tokenizers_deps INTERFACE "${TOKENIZERS_DIR}")
-
-# Source Definitions
-set(CORE_SOURCES
-    ${RNEXECUTORCH_DIR}/models/BaseModel.cpp
-    ${RNEXECUTORCH_DIR}/data_processing/Numerical.cpp
-    ${CMAKE_SOURCE_DIR}/integration/stubs/jsi_stubs.cpp
-)
-
-set(IMAGE_UTILS_SOURCES
-    ${RNEXECUTORCH_DIR}/data_processing/ImageProcessing.cpp
-    ${RNEXECUTORCH_DIR}/data_processing/base64.cpp
-    ${COMMON_DIR}/ada/ada.cpp
-)
-
-set(TOKENIZER_SOURCES ${RNEXECUTORCH_DIR}/TokenizerModule.cpp)
-set(DSP_SOURCES ${RNEXECUTORCH_DIR}/data_processing/dsp.cpp)
-
-# Core Library
-add_library(rntests_core STATIC ${CORE_SOURCES})
-
-target_include_directories(rntests_core PUBLIC
-    ${RNEXECUTORCH_DIR}/data_processing
-    ${TOKENIZERS_DIR}
-    ${RNEXECUTORCH_DIR}
-    ${COMMON_DIR}
-    ${PACKAGE_ROOT}/third-party/include
-    ${REACT_NATIVE_DIR}/ReactCommon
-    ${REACT_NATIVE_DIR}/ReactCommon/jsi
-    ${REACT_NATIVE_DIR}/ReactCommon/callinvoker
-    ${COMMON_DIR}/ada
-)
-
-target_link_libraries(rntests_core PUBLIC
-    executorch_prebuilt
-    gtest
-    log
-)
-
-enable_testing()
-function(add_rn_test TEST_TARGET TEST_FILENAME)
-    cmake_parse_arguments(ARG "" "" "SOURCES;LIBS" ${ARGN})
-    # Create executable using the explicit filename provided
-    add_executable(${TEST_TARGET} ${TEST_FILENAME} ${ARG_SOURCES})
-
-    target_compile_definitions(${TEST_TARGET} PRIVATE TEST_BUILD)
-    target_link_libraries(${TEST_TARGET} PRIVATE rntests_core gtest_main ${ARG_LIBS})
-    target_link_options(${TEST_TARGET} PRIVATE "LINKER:-z,max-page-size=16384")
-
-    add_test(NAME ${TEST_TARGET} COMMAND ${TEST_TARGET})
-endfunction()
-
-add_rn_test(NumericalTests unit/NumericalTest.cpp)
-add_rn_test(LogTests unit/LogTest.cpp)
-add_rn_test(BaseModelTests integration/BaseModelTest.cpp)
-
-add_rn_test(ClassificationTests integration/ClassificationTest.cpp
-    SOURCES
-        ${RNEXECUTORCH_DIR}/models/classification/Classification.cpp
-        ${IMAGE_UTILS_SOURCES}
-    LIBS opencv_deps
-)
-
-add_rn_test(ObjectDetectionTests integration/ObjectDetectionTest.cpp
-    SOURCES
-        ${RNEXECUTORCH_DIR}/models/object_detection/ObjectDetection.cpp
-        ${RNEXECUTORCH_DIR}/models/object_detection/Utils.cpp
-        ${IMAGE_UTILS_SOURCES}
-    LIBS opencv_deps
-)
-
-add_rn_test(ImageEmbeddingsTests integration/ImageEmbeddingsTest.cpp
-    SOURCES
-        ${RNEXECUTORCH_DIR}/models/embeddings/image/ImageEmbeddings.cpp
-        ${RNEXECUTORCH_DIR}/models/embeddings/BaseEmbeddings.cpp
-        ${IMAGE_UTILS_SOURCES}
-    LIBS opencv_deps
-)
-
-add_rn_test(TextEmbeddingsTests integration/TextEmbeddingsTest.cpp
-    SOURCES
-        ${RNEXECUTORCH_DIR}/models/embeddings/text/TextEmbeddings.cpp
-        ${RNEXECUTORCH_DIR}/models/embeddings/BaseEmbeddings.cpp
-        ${TOKENIZER_SOURCES}
-    LIBS tokenizers_deps
-)
-
-add_rn_test(StyleTransferTests integration/StyleTransferTest.cpp
-    SOURCES
-        ${RNEXECUTORCH_DIR}/models/style_transfer/StyleTransfer.cpp
-        ${IMAGE_UTILS_SOURCES}
-    LIBS opencv_deps
-)
-
-add_rn_test(VADTests integration/VoiceActivityDetectionTest.cpp
-    SOURCES
-        ${RNEXECUTORCH_DIR}/models/voice_activity_detection/VoiceActivityDetection.cpp
-        ${RNEXECUTORCH_DIR}/models/voice_activity_detection/Utils.cpp
-        ${DSP_SOURCES}
-)
-
-add_rn_test(TokenizerModuleTests integration/TokenizerModuleTest.cpp
-    SOURCES ${TOKENIZER_SOURCES}
-    LIBS tokenizers_deps
-)
-
-add_rn_test(SpeechToTextTests integration/SpeechToTextTest.cpp
-    SOURCES
-        ${RNEXECUTORCH_DIR}/models/speech_to_text/SpeechToText.cpp
-        ${RNEXECUTORCH_DIR}/models/speech_to_text/asr/ASR.cpp
-        ${RNEXECUTORCH_DIR}/models/speech_to_text/stream/HypothesisBuffer.cpp
-        ${RNEXECUTORCH_DIR}/models/speech_to_text/stream/OnlineASRProcessor.cpp
-        ${RNEXECUTORCH_DIR}/data_processing/gzip.cpp
-        ${TOKENIZER_SOURCES}
-        ${DSP_SOURCES}
-    LIBS tokenizers_deps z
-)
-
-add_rn_test(LLMTests integration/LLMTest.cpp
-    SOURCES
-        ${RNEXECUTORCH_DIR}/models/llm/LLM.cpp
-        ${COMMON_DIR}/runner/runner.cpp
-        ${COMMON_DIR}/runner/text_prefiller.cpp
-        ${COMMON_DIR}/runner/text_decoder_runner.cpp
-        ${COMMON_DIR}/runner/sampler.cpp
-        ${COMMON_DIR}/runner/arange_util.cpp
-    LIBS tokenizers_deps
-)
-
-add_rn_test(TextToImageTests integration/TextToImageTest.cpp
-    SOURCES
-        ${RNEXECUTORCH_DIR}/models/text_to_image/TextToImage.cpp
-        ${RNEXECUTORCH_DIR}/models/text_to_image/Encoder.cpp
-        ${RNEXECUTORCH_DIR}/models/text_to_image/UNet.cpp
-        ${RNEXECUTORCH_DIR}/models/text_to_image/Decoder.cpp
-        ${RNEXECUTORCH_DIR}/models/text_to_image/Scheduler.cpp
-        ${RNEXECUTORCH_DIR}/models/embeddings/text/TextEmbeddings.cpp
-        ${RNEXECUTORCH_DIR}/models/embeddings/BaseEmbeddings.cpp
-        ${TOKENIZER_SOURCES}
-    LIBS tokenizers_deps
-)
-
-add_rn_test(OCRTests integration/OCRTest.cpp
-    SOURCES
-        ${RNEXECUTORCH_DIR}/models/ocr/OCR.cpp
-        ${RNEXECUTORCH_DIR}/models/ocr/CTCLabelConverter.cpp
-        ${RNEXECUTORCH_DIR}/models/ocr/Detector.cpp
-        ${RNEXECUTORCH_DIR}/models/ocr/RecognitionHandler.cpp
-        ${RNEXECUTORCH_DIR}/models/ocr/Recognizer.cpp
-        ${RNEXECUTORCH_DIR}/models/ocr/utils/DetectorUtils.cpp
-        ${RNEXECUTORCH_DIR}/models/ocr/utils/RecognitionHandlerUtils.cpp
-        ${RNEXECUTORCH_DIR}/models/ocr/utils/RecognizerUtils.cpp
-        ${IMAGE_UTILS_SOURCES}
-    LIBS opencv_deps
-)
-
-add_rn_test(VerticalOCRTests integration/VerticalOCRTest.cpp
-    SOURCES
-        ${RNEXECUTORCH_DIR}/models/vertical_ocr/VerticalOCR.cpp
-        ${RNEXECUTORCH_DIR}/models/vertical_ocr/VerticalDetector.cpp
-        ${RNEXECUTORCH_DIR}/models/ocr/Detector.cpp
-        ${RNEXECUTORCH_DIR}/models/ocr/CTCLabelConverter.cpp
-        ${RNEXECUTORCH_DIR}/models/ocr/Recognizer.cpp
-        ${RNEXECUTORCH_DIR}/models/ocr/utils/DetectorUtils.cpp
-        ${RNEXECUTORCH_DIR}/models/ocr/utils/RecognitionHandlerUtils.cpp
-        ${RNEXECUTORCH_DIR}/models/ocr/utils/RecognizerUtils.cpp
-        ${IMAGE_UTILS_SOURCES}
-    LIBS opencv_deps
-)

package/common/rnexecutorch/tests/README.md

@@ -1,73 +0,0 @@
-## Native Test
-This guide provide information on how functions are tested, how to install all needed dependencies and how to run tests.
-
-### Used Tools
-To test the native code we use [`googletest`](https://github.com/google/googletest). It's a flexible tool for creating unit tests.
-
-### Installation
-The googletest is already in repo in `react-native-executorch/third-party/googletest`. Firstly, you need to fetch googletest locally, run from root directory of project:
-* `git submodule update --init --recursive third-party/googletest`
-
-### Running tests
-
-#### Prerequisites
-
-- **Android NDK**: The `ANDROID_NDK` environment variable must be set
-- **wget**: Must be in your PATH
-- **Android emulator**: Must be running before executing tests
-- **Device requirements**:
-  - 16GB disk storage (minimum)
-  - 8GB RAM (minimum)
-
-#### First-time setup
-
-Before running tests, you need to build an app to generate required native libraries (`libfbjni.so` and `libc++_shared.so`). The test script automatically searches for these in the monorepo.
-
-If the script reports missing libraries, build any example app:
-```bash
-cd apps/computer-vision/android
-./gradlew assembleDebug
-# or
-./gradlew assembleRelease
-```
-
-#### Running the tests
-
-Navigate to the tests directory:
-```bash
-cd packages/react-native-executorch/common/rnexecutorch/tests
-```
-
-Run the test script:
-```bash
-bash ./run_tests.sh
-```
-
-This script:
-- Downloads all needed models
-- Pushes executables, models, assets, and shared libraries via ADB to the running emulator
-- Runs the pre-compiled test executables
-
-#### Available flags
-
-* `--refresh-models` - Forcefully downloads all the models. By default, models are not downloaded unless they are missing from the specified directory.
-* `--skip-build` - Skips the cmake build step.
-
-### How to add a new test
-To add new test you need to:
-* Add a new .cpp file to either integration/ or unit/, depending on the type of the test.
-* In `CMakeLists.txt`, add all executables and link all the needed libraries against the executable, for example you can use the `add_rn_test`, which is a helper function that links core libs. Example:
-    ```cmake
-    # unit
-    add_rn_test(BaseModelTests integration/BaseModelTest.cpp)
-
-    # integration
-    add_rn_test(ClassificationTests integration/ClassificationTest.cpp
-        SOURCES
-            ${RNEXECUTORCH_DIR}/models/classification/Classification.cpp
-            ${IMAGE_UTILS_SOURCES}
-        LIBS opencv_deps
-    )
-    ```
-* Lastly, add the test executable name to the run_tests script along with all the needed URL and assets.
-

package/common/rnexecutorch/tests/integration/BaseModelTest.cpp

@@ -1,207 +0,0 @@
-#include "BaseModelTests.h"
-#include <executorch/extension/tensor/tensor.h>
-#include <gtest/gtest.h>
-#include <limits>
-#include <rnexecutorch/Error.h>
-#include <rnexecutorch/models/BaseModel.h>
-#include <vector>
-
-using namespace rnexecutorch;
-using namespace rnexecutorch::models;
-using namespace executorch::extension;
-using namespace model_tests;
-using executorch::runtime::EValue;
-
-constexpr auto kValidStyleTransferModelPath =
-    "style_transfer_candy_xnnpack.pte";
-
-// ============================================================================
-// Common tests via typed test suite
-// ============================================================================
-namespace model_tests {
-template <> struct ModelTraits<BaseModel> {
-  using ModelType = BaseModel;
-
-  static ModelType createValid() {
-    return ModelType(kValidStyleTransferModelPath, nullptr);
-  }
-
-  static ModelType createInvalid() {
-    return ModelType("nonexistent.pte", nullptr);
-  }
-
-  static void callGenerate(ModelType &model) {
-    std::vector<int32_t> shape = {1, 3, 640, 640};
-    size_t numElements = 1 * 3 * 640 * 640;
-    std::vector<float> inputData(numElements, 0.5f);
-    auto tensorPtr = make_tensor_ptr(shape, inputData.data());
-    EValue input(*tensorPtr);
-    (void)model.forward(input);
-  }
-};
-} // namespace model_tests
-
-using BaseModelTypes = ::testing::Types<BaseModel>;
-INSTANTIATE_TYPED_TEST_SUITE_P(BaseModel, CommonModelTest, BaseModelTypes);
-
-// ============================================================================
-// BaseModel-specific tests (methods not in all models)
-// ============================================================================
-
-TEST(BaseModelGetInputShapeTests, ValidMethodCorrectShape) {
-  BaseModel model(kValidStyleTransferModelPath, nullptr);
-  auto forwardShape = model.getInputShape("forward", 0);
-  std::vector<int32_t> expectedShape = {1, 3, 640, 640};
-  EXPECT_EQ(forwardShape, expectedShape);
-}
-
-TEST(BaseModelGetInputShapeTests, InvalidMethodThrows) {
-  BaseModel model(kValidStyleTransferModelPath, nullptr);
-  EXPECT_THROW((void)model.getInputShape("this_method_does_not_exist", 0),
-               RnExecutorchError);
-}
-
-TEST(BaseModelGetInputShapeTests, ValidMethodInvalidIndexThrows) {
-  BaseModel model(kValidStyleTransferModelPath, nullptr);
-  EXPECT_THROW(
-      (void)model.getInputShape("forward", std::numeric_limits<int32_t>::min()),
-      RnExecutorchError);
-}
-
-TEST(BaseModelGetAllInputShapesTests, ValidMethodReturnsShapes) {
-  BaseModel model(kValidStyleTransferModelPath, nullptr);
-  auto allShapes = model.getAllInputShapes("forward");
-  EXPECT_FALSE(allShapes.empty());
-  std::vector<int32_t> expectedFirstShape = {1, 3, 640, 640};
-  EXPECT_EQ(allShapes[0], expectedFirstShape);
-}
-
-TEST(BaseModelGetAllInputShapesTests, InvalidMethodThrows) {
-  BaseModel model(kValidStyleTransferModelPath, nullptr);
-  EXPECT_THROW(model.getAllInputShapes("non_existent_method"),
-               RnExecutorchError);
-}
-
-TEST(BaseModelGetMethodMetaTests, ValidMethodReturnsOk) {
-  BaseModel model(kValidStyleTransferModelPath, nullptr);
-  auto result = model.getMethodMeta("forward");
-  EXPECT_TRUE(result.ok());
-}
-
-TEST(BaseModelGetMethodMetaTests, InvalidMethodReturnsError) {
-  BaseModel model(kValidStyleTransferModelPath, nullptr);
-  auto result = model.getMethodMeta("non_existent_method");
-  EXPECT_FALSE(result.ok());
-}
-
-TEST(BaseModelForwardTests, ForwardWithValidInputReturnsOk) {
-  BaseModel model(kValidStyleTransferModelPath, nullptr);
-  std::vector<int32_t> shape = {1, 3, 640, 640};
-  size_t numElements = 1 * 3 * 640 * 640;
-  std::vector<float> inputData(numElements, 0.5f);
-  auto tensorPtr = make_tensor_ptr(shape, inputData.data());
-  EValue input(*tensorPtr);
-
-  auto result = model.forward(input);
-  EXPECT_TRUE(result.ok());
-}
-
-TEST(BaseModelForwardTests, ForwardWithVectorInputReturnsOk) {
-  BaseModel model(kValidStyleTransferModelPath, nullptr);
-  std::vector<int32_t> shape = {1, 3, 640, 640};
-  size_t numElements = 1 * 3 * 640 * 640;
-  std::vector<float> inputData(numElements, 0.5f);
-  auto tensorPtr = make_tensor_ptr(shape, inputData.data());
-  std::vector<EValue> inputs;
-  inputs.emplace_back(*tensorPtr);
-
-  auto result = model.forward(inputs);
-  EXPECT_TRUE(result.ok());
-}
-
-TEST(BaseModelForwardTests, ForwardReturnsCorrectOutputShape) {
-  BaseModel model(kValidStyleTransferModelPath, nullptr);
-  std::vector<int32_t> shape = {1, 3, 640, 640};
-  size_t numElements = 1 * 3 * 640 * 640;
-  std::vector<float> inputData(numElements, 0.5f);
-  auto tensorPtr = make_tensor_ptr(shape, inputData.data());
-  EValue input(*tensorPtr);
-
-  auto result = model.forward(input);
-  ASSERT_TRUE(result.ok());
-  ASSERT_FALSE(result->empty());
-
-  auto &outputTensor = result->at(0).toTensor();
-  auto outputSizes = outputTensor.sizes();
-  EXPECT_EQ(outputSizes.size(), 4);
-  EXPECT_EQ(outputSizes[0], 1);
-  EXPECT_EQ(outputSizes[1], 3);
-  EXPECT_EQ(outputSizes[2], 640);
-  EXPECT_EQ(outputSizes[3], 640);
-}
-
-TEST(BaseModelForwardTests, ForwardAfterUnloadThrows) {
-  BaseModel model(kValidStyleTransferModelPath, nullptr);
-  model.unload();
-
-  std::vector<int32_t> shape = {1, 3, 640, 640};
-  size_t numElements = 1 * 3 * 640 * 640;
-  std::vector<float> inputData(numElements, 0.5f);
-  auto tensorPtr = make_tensor_ptr(shape, inputData.data());
-  EValue input(*tensorPtr);
-
-  EXPECT_THROW(model.forward(input), RnExecutorchError);
-}
-
-TEST(BaseModelForwardJSTests, ForwardJSWithValidInputReturnsOutput) {
-  BaseModel model(kValidStyleTransferModelPath, nullptr);
-  std::vector<int32_t> shape = {1, 3, 640, 640};
-  size_t numElements = 1 * 3 * 640 * 640;
-  std::vector<float> inputData(numElements, 0.5f);
-
-  JSTensorViewIn tensorView;
-  tensorView.dataPtr = inputData.data();
-  tensorView.sizes = shape;
-  tensorView.scalarType = executorch::aten::ScalarType::Float;
-
-  std::vector<JSTensorViewIn> inputs = {tensorView};
-  auto outputs = model.forwardJS(inputs);
-
-  EXPECT_FALSE(outputs.empty());
-}
-
-TEST(BaseModelForwardJSTests, ForwardJSReturnsCorrectOutputShape) {
-  BaseModel model(kValidStyleTransferModelPath, nullptr);
-  std::vector<int32_t> shape = {1, 3, 640, 640};
-  size_t numElements = 1 * 3 * 640 * 640;
-  std::vector<float> inputData(numElements, 0.5f);
-
-  JSTensorViewIn tensorView;
-  tensorView.dataPtr = inputData.data();
-  tensorView.sizes = shape;
-  tensorView.scalarType = executorch::aten::ScalarType::Float;
-
-  std::vector<JSTensorViewIn> inputs = {tensorView};
-  auto outputs = model.forwardJS(inputs);
-
-  ASSERT_EQ(outputs.size(), 1);
-  std::vector<int32_t> expectedShape = {1, 3, 640, 640};
-  EXPECT_EQ(outputs[0].sizes, expectedShape);
-}
-
-TEST(BaseModelForwardJSTests, ForwardJSAfterUnloadThrows) {
-  BaseModel model(kValidStyleTransferModelPath, nullptr);
-  model.unload();
-
-  std::vector<int32_t> shape = {1, 3, 640, 640};
-  size_t numElements = 1 * 3 * 640 * 640;
-  std::vector<float> inputData(numElements, 0.5f);
-
-  JSTensorViewIn tensorView;
-  tensorView.dataPtr = inputData.data();
-  tensorView.sizes = shape;
-  tensorView.scalarType = executorch::aten::ScalarType::Float;
-
-  std::vector<JSTensorViewIn> inputs = {tensorView};
-  EXPECT_THROW((void)model.forwardJS(inputs), RnExecutorchError);
-}

package/common/rnexecutorch/tests/integration/BaseModelTests.h

@@ -1,120 +0,0 @@
-#pragma once
-
-#include "gtest/gtest.h"
-#include <rnexecutorch/Error.h>
-
-namespace facebook::react {
-class CallInvoker;
-}
-
-namespace rnexecutorch {
-std::shared_ptr<facebook::react::CallInvoker> createMockCallInvoker();
-}
-
-namespace model_tests {
-
-inline auto getMockInvoker() { return rnexecutorch::createMockCallInvoker(); }
-
-/// Helper macro to access Traits in typed tests
-#define SETUP_TRAITS() using Traits = typename TestFixture::Traits
-
-/// Trait struct that each model must specialize
-/// This defines how to construct and test each model type
-template <typename T> struct ModelTraits;
-
-/// Example of what a specialization looks like:
-///
-/// template<>
-/// struct ModelTraits<Classification> {
-///   using ModelType = Classification;
-///
-///   // Create valid model instance
-///   static ModelType createValid() {
-///     return ModelType("valid_model.pte", nullptr);
-///   }
-///
-///   // Create invalid model instance (should throw in constructor)
-///   static ModelType createInvalid() {
-///     return ModelType("nonexistent.pte", nullptr);
-///   }
-///
-///   // Call the model's generate/forward function with valid input
-///   // Used to test that generate throws after unload
-///  static void callGenerate(ModelType& model) {
-///     (void)model.generate("valid_input.jpg");
-///   }
-/// };
-// Typed test fixture for common model tests
-template <typename T> class CommonModelTest : public ::testing::Test {
-protected:
-  using Traits = ModelTraits<T>;
-  using ModelType = typename Traits::ModelType;
-};
-
-// Define the test suite
-TYPED_TEST_SUITE_P(CommonModelTest);
-
-// Constructor tests
-TYPED_TEST_P(CommonModelTest, InvalidPathThrows) {
-  SETUP_TRAITS();
-  EXPECT_THROW(Traits::createInvalid(), rnexecutorch::RnExecutorchError);
-}
-
-TYPED_TEST_P(CommonModelTest, ValidPathDoesntThrow) {
-  SETUP_TRAITS();
-  EXPECT_NO_THROW(Traits::createValid());
-}
-
-// Memory tests
-TYPED_TEST_P(CommonModelTest, GetMemoryLowerBoundValue) {
-  SETUP_TRAITS();
-  auto model = Traits::createValid();
-  EXPECT_GT(model.getMemoryLowerBound(), 0u);
-}
-
-TYPED_TEST_P(CommonModelTest, GetMemoryLowerBoundConsistent) {
-  SETUP_TRAITS();
-  auto model = Traits::createValid();
-  auto bound1 = model.getMemoryLowerBound();
-  auto bound2 = model.getMemoryLowerBound();
-  EXPECT_EQ(bound1, bound2);
-}
-
-// Unload tests
-TYPED_TEST_P(CommonModelTest, UnloadDoesntThrow) {
-  SETUP_TRAITS();
-  auto model = Traits::createValid();
-  EXPECT_NO_THROW(model.unload());
-}
-
-TYPED_TEST_P(CommonModelTest, MultipleUnloadsSafe) {
-  SETUP_TRAITS();
-  auto model = Traits::createValid();
-  EXPECT_NO_THROW(model.unload());
-  EXPECT_NO_THROW(model.unload());
-  EXPECT_NO_THROW(model.unload());
-}
-
-TYPED_TEST_P(CommonModelTest, GenerateAfterUnloadThrows) {
-  SETUP_TRAITS();
-  auto model = Traits::createValid();
-  model.unload();
-  EXPECT_THROW(Traits::callGenerate(model), rnexecutorch::RnExecutorchError);
-}
-
-TYPED_TEST_P(CommonModelTest, MultipleGeneratesWork) {
-  SETUP_TRAITS();
-  auto model = Traits::createValid();
-  EXPECT_NO_THROW(Traits::callGenerate(model));
-  EXPECT_NO_THROW(Traits::callGenerate(model));
-  EXPECT_NO_THROW(Traits::callGenerate(model));
-}
-
-// Register all tests in the suite
-REGISTER_TYPED_TEST_SUITE_P(CommonModelTest, InvalidPathThrows,
-                            ValidPathDoesntThrow, GetMemoryLowerBoundValue,
-                            GetMemoryLowerBoundConsistent, UnloadDoesntThrow,
-                            MultipleUnloadsSafe, GenerateAfterUnloadThrows,
-                            MultipleGeneratesWork);
-
-} // namespace model_tests