Add/Get n-gram probability entry in languageModelDictContent

Bug: 14425059
Change-Id: I7926c3812f89b9a71fe1873a5bc32f793f91b640

1 files changed, 12 insertions, 3 deletions

diff --git a/native/jni/tests/utils/int_array_view_test.cpp b/native/jni/tests/utils/int_array_view_test.cpp
index 9aa8cdc8c..f799dd88f 100644
--- a/native/jni/tests/utils/int_array_view_test.cpp
+++ b/native/jni/tests/utils/int_array_view_test.cpp
@@ -24,15 +24,24 @@ namespace latinime {
 namespace {
 
 TEST(MemoryViewTest, TestAccess) {
-    static const int DATA_SIZE = 10000;
-
     std::vector<int> intVector = {3, 2, 1, 0, -1, -2};
     IntArrayView intArrayView(intVector);
     EXPECT_EQ(intVector.size(), intArrayView.size());
-    for (int i = 0; i < DATA_SIZE; ++i) {
+    for (int i = 0; i < static_cast<int>(intVector.size()); ++i) {
         EXPECT_EQ(intVector[i], intArrayView[i]);
     }
 }
 
+TEST(MemoryViewTest, TestIteration) {
+    std::vector<int> intVector = {3, 2, 1, 0, -1, -2};
+    IntArrayView intArrayView(intVector);
+    std::set<int> intSet(intVector.begin(), intVector.end());
+    for (const int i : intArrayView) {
+        EXPECT_TRUE(intSet.count(i) > 0);
+        intSet.erase(i);
+    }
+    EXPECT_TRUE(intSet.empty());
+}
+
 }  // namespace
 }  // namespace latinime