From eaf6edc3a6c95fe270790bb59e308d53887aa1a8 Mon Sep 17 00:00:00 2001
From: Prianka Liz Kariat <prianka.kariat@ymedialabs.com>
Date: Thu, 1 Dec 2022 09:13:45 +0530
Subject: [PATCH] Added Objc tests for iOS Text Classifier

---
 .../tasks/ios/test/text/text_classifier/BUILD | 36 ++++++++++++
 .../text_classifier/MPPTextClassifierTests.m  | 57 +++++++++++++++++++
 2 files changed, 93 insertions(+)
 create mode 100644 mediapipe/tasks/ios/test/text/text_classifier/BUILD
 create mode 100644 mediapipe/tasks/ios/test/text/text_classifier/MPPTextClassifierTests.m

diff --git a/mediapipe/tasks/ios/test/text/text_classifier/BUILD b/mediapipe/tasks/ios/test/text/text_classifier/BUILD
new file mode 100644
index 000000000..cbcc3d106
--- /dev/null
+++ b/mediapipe/tasks/ios/test/text/text_classifier/BUILD
@@ -0,0 +1,36 @@
+load("@org_tensorflow//tensorflow/lite/ios:ios.bzl", "TFL_DEFAULT_TAGS", "TFL_DISABLED_SANITIZER_TAGS", "TFL_MINIMUM_OS_VERSION")
+load("@build_bazel_rules_apple//apple:ios.bzl", "ios_unit_test")
+load("@org_tensorflow//tensorflow/lite:special_rules.bzl", "tflite_ios_lab_runner")
+
+package(default_visibility = ["//mediapipe/tasks:internal"])
+
+licenses(["notice"])
+
+objc_library(
+    name = "MPPTextClassifierObjcTestLibrary",
+    testonly = 1,
+    srcs = ["MPPTextClassifierTests.m"],
+    data = [
+        "//mediapipe/tasks/testdata/text:bert_text_classifier_models",
+        "//mediapipe/tasks/testdata/text:text_classifier_models",
+    ],
+    tags = TFL_DEFAULT_TAGS,
+    copts = [
+        "-ObjC++",
+        "-std=c++17",
+    ],
+    deps = [
+        "//mediapipe/tasks/ios/text/text_classifier:MPPTextClassifier",
+    ],
+    
+)
+
+ios_unit_test(
+    name = "MPPTextClassifierObjcTest",
+    minimum_os_version = TFL_MINIMUM_OS_VERSION,
+    runner = tflite_ios_lab_runner("IOS_LATEST"),
+    tags = TFL_DEFAULT_TAGS + TFL_DISABLED_SANITIZER_TAGS,
+    deps = [
+        ":MPPTextClassifierObjcTestLibrary",
+    ],
+)
diff --git a/mediapipe/tasks/ios/test/text/text_classifier/MPPTextClassifierTests.m b/mediapipe/tasks/ios/test/text/text_classifier/MPPTextClassifierTests.m
new file mode 100644
index 000000000..3808009f3
--- /dev/null
+++ b/mediapipe/tasks/ios/test/text/text_classifier/MPPTextClassifierTests.m
@@ -0,0 +1,57 @@
+/* Copyright 2022 The TensorFlow Authors. All Rights Reserved.
+
+ Licensed under the Apache License, Version 2.0 (the "License");
+ you may not use this file except in compliance with the License.
+ You may obtain a copy of the License at
+
+ http://www.apache.org/licenses/LICENSE-2.0
+
+ Unless required by applicable law or agreed to in writing, software
+ distributed under the License is distributed on an "AS IS" BASIS,
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ See the License for the specific language governing permissions and
+ limitations under the License.
+ ==============================================================================*/
+#import <XCTest/XCTest.h>
+
+#import "mediapipe/tasks/ios/text/text_classifier/sources/MPPTextClassifier.h"
+
+NS_ASSUME_NONNULL_BEGIN
+
+static NSString *const kBertTextClassifierModelName = @"bert_text_classifier";
+
+@interface MPPTextClassifierTests : XCTestCase
+@end
+
+@implementation MPPTextClassifierTests
+
+- (void)setUp {
+  [super setUp];
+
+}
+
+- (NSString *)filePathWithName:(NSString *)fileName extension:(NSString *)extension {
+  NSString *filePath = [[NSBundle bundleForClass:self.class] pathForResource:fileName
+                                                                      ofType:extension];
+  XCTAssertNotNil(filePath);
+
+  return filePath;
+}
+
+- (MPPTextClassifierOptions *)textClassifierOptionsWithModelName:(NSString *)modelName {
+  NSString *modelPath = [self filePathWithName:modelName extension:@"tflite"];
+  MPPTextClassifierOptions *textClassifierOptions =
+      [[MPPTextClassifierOptions alloc] initWithModelPath:modelPath];
+
+  return textClassifierOptions;
+}
+
+- (void)testCreateTextClassifierOptionsSucceeds {
+  MPPTextClassifierOptions *options = [self textClassifierOptionsWithModelName:kBertTextClassifierModelName];
+  MPPTextClassifier *textClassifier = [[MPPTextClassifier alloc] initWithOptions:options error:nil];
+  XCTAssertNotNil(textClassifier);
+}
+
+@end
+
+NS_ASSUME_NONNULL_END