Skip to content

Added iOS Text Searcher and Basic Objective C Test #898

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 6 commits into
base: master
Choose a base branch
from
Open

Added iOS Text Searcher and Basic Objective C Test #898

Changes from 1 commit
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
41c6518
Added text searcher files
priankakariat Oct 20, 2022
4ea863e
Merge branch 'master' into ios-text-searcher
priankakariat Oct 20, 2022
e9c0568
Updated text searcher tests
priankakariat Oct 20, 2022
7635d7f
Merge branch 'master' into ios-text-searcher
priankakariat Nov 2, 2022
0ae27a0
Updated tests
priankakariat Nov 2, 2022
6760d80
Added variable for string
priankakariat Nov 2, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Updated text searcher tests
  • Loading branch information
@priankakariat
priankakariat committed Oct 20, 2022
commit e9c05680e1b35d5dfb0bd02f4b44e60257f1fb28
36 changes: 18 additions & 18 deletions tensorflow_lite_support/ios/test/task/text/text_searcher/TFLTextSearcherTests.m
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,10 +18,10 @@

NS_ASSUME_NONNULL_BEGIN

#define VerifySearchResultCount(searchResult, expectedNearestNeighborsCount) \
#define ValidateSearchResultCount(searchResult, expectedNearestNeighborsCount) \
XCTAssertEqual(searchResult.nearestNeighbors.count, expectedNearestNeighborsCount);

#define VerifyNearestNeighbor(nearestNeighbor, expectedMetadata, expectedDistance) \
#define ValidateNearestNeighbor(nearestNeighbor, expectedMetadata, expectedDistance) \
XCTAssertEqualObjects(nearestNeighbor.metadata, expectedMetadata); \
XCTAssertEqualWithAccuracy(nearestNeighbor.distance, expectedDistance, 1e-6);

Expand Down Expand Up @@ -51,29 +51,29 @@ - (TFLTextSearcher *)testSuccessfulCreationOfTextSearcherWithSearchContent:(NSSt
}

- (void)verifySearchResultForInferenceWithSearchContent:(TFLSearchResult *)searchResult {
VerifySearchResultCount(searchResult,
ValidateSearchResultCount(searchResult,
5 // expectedNearestNeighborsCount
);

VerifyNearestNeighbor(searchResult.nearestNeighbors[0],
@"burger", // expectedMetadata
198.456329 // expectedDistance
ValidateNearestNeighbor(searchResult.nearestNeighbors[0],
@"The weather was excellent.", // expectedMetadata
0.889664649963 // expectedDistance
);
VerifyNearestNeighbor(searchResult.nearestNeighbors[1],
@"car", // expectedMetadata
226.022186 // expectedDistance
ValidateNearestNeighbor(searchResult.nearestNeighbors[1],
@"The sun was shining on that day.", // expectedMetadata
0.889667928219 // expectedDistance
);
VerifyNearestNeighbor(searchResult.nearestNeighbors[2],
@"bird", // expectedMetadata
227.297668 // expectedDistance
ValidateNearestNeighbor(searchResult.nearestNeighbors[2],
@"The cat is chasing after the mouse.", // expectedMetadata
0.889669716358 // expectedDistance
);
VerifyNearestNeighbor(searchResult.nearestNeighbors[3],
@"dog", // expectedMetadata
229.133789 // expectedDistance
ValidateNearestNeighbor(searchResult.nearestNeighbors[3],
@"It was a sunny day.", // expectedMetadata
0.889671087265 // expectedDistance
);
VerifyNearestNeighbor(searchResult.nearestNeighbors[4],
@"cat", // expectedMetadata
229.718948 // expectedDistance
ValidateNearestNeighbor(searchResult.nearestNeighbors[4],
@"He was very happy with his newly bought car.", // expectedMetadata
0.889671683311 // expectedDistance
);
}

Expand Down