cassdlcm/mediapipe
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Updated method names in tests

Browse Source
This commit is contained in:
Prianka Liz Kariat 2023-01-16 13:05:29 +05:30
parent 8ecf77f760
commit f7fc8a6eca
2 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
mediapipe/tasks/ios/test/text/text_classifier/MPPTextClassifierTests.m
View File

@ -43,7 +43,6 @@ static NSString *const kExpectedErrorDomain = @"com.google.mediapipe.tasks";
#define AssertTextClassifierResultHasOneHead(textClassifierResult) \ #define AssertTextClassifierResultHasOneHead(textClassifierResult) \
XCTAssertNotNil(textClassifierResult); \ XCTAssertNotNil(textClassifierResult); \
\
XCTAssertNotNil(textClassifierResult.classificationResult); \ XCTAssertNotNil(textClassifierResult.classificationResult); \
XCTAssertEqual(textClassifierResult.classificationResult.classifications.count, 1); \ XCTAssertEqual(textClassifierResult.classificationResult.classifications.count, 1); \
XCTAssertEqual(textClassifierResult.classificationResult.classifications[0].headIndex, 0); XCTAssertEqual(textClassifierResult.classificationResult.classifications[0].headIndex, 0);
@ -156,7 +155,7 @@ static NSString *const kExpectedErrorDomain = @"com.google.mediapipe.tasks";
AssertEqualErrors(error, expectedError); AssertEqualErrors(error, expectedError);
} }
- (void)testCreateTextClassifierFailsWithBothAllowListAndDenyList { - (void)testCreateTextClassifierFailsWithBothAllowlistAndDenylist {
MPPTextClassifierOptions *options = MPPTextClassifierOptions *options =
[self textClassifierOptionsWithModelName:kBertTextClassifierModelName]; [self textClassifierOptionsWithModelName:kBertTextClassifierModelName];
options.categoryAllowlist = @[ @"positive" ]; options.categoryAllowlist = @[ @"positive" ];
@ -233,7 +232,7 @@ static NSString *const kExpectedErrorDomain = @"com.google.mediapipe.tasks";
expectedBertResultCategoriesForEdgeCaseTests]]; expectedBertResultCategoriesForEdgeCaseTests]];
} }
- (void)testClassifyWithCategoryAllowListSucceeds { - (void)testClassifyWithCategoryAllowlistSucceeds {
MPPTextClassifierOptions *options = MPPTextClassifierOptions *options =
[self textClassifierOptionsWithModelName:kBertTextClassifierModelName]; [self textClassifierOptionsWithModelName:kBertTextClassifierModelName];
options.categoryAllowlist = @[ @"negative" ]; options.categoryAllowlist = @[ @"negative" ];
@ -250,7 +249,7 @@ static NSString *const kExpectedErrorDomain = @"com.google.mediapipe.tasks";
expectedBertResultCategoriesForEdgeCaseTests]]; expectedBertResultCategoriesForEdgeCaseTests]];
} }
- (void)testClassifyWithCategoryDenyListSucceeds { - (void)testClassifyWithCategoryDenylistSucceeds {
MPPTextClassifierOptions *options = MPPTextClassifierOptions *options =
[self textClassifierOptionsWithModelName:kBertTextClassifierModelName]; [self textClassifierOptionsWithModelName:kBertTextClassifierModelName];
options.categoryDenylist = @[ @"positive" ]; options.categoryDenylist = @[ @"positive" ];

2
mediapipe/tasks/ios/test/text/text_classifier/TextClassifierTests.swift
View File

@ -147,7 +147,7 @@ class TextClassifierTests: XCTestCase {
""") """)
} }
func testCreateTextClassifierWithCategoryAllowlistandDenylistFails() throws { func testCreateTextClassifierWithCategoryAllowlistAndDenylistFails() throws {
let textClassifierOptions = let textClassifierOptions =
try XCTUnwrap( try XCTUnwrap(