| /* |
| * Copyright (C) 2014 The Android Open Source Project |
| * |
| * 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. |
| */ |
| |
| #include "suggest/policyimpl/dictionary/structure/v4/content/probability_entry.h" |
| |
| #include <gtest/gtest.h> |
| |
| #include "defines.h" |
| |
| namespace latinime { |
| namespace { |
| |
| TEST(ProbabilityEntryTest, TestEncodeDecode) { |
| const int flag = 0xFF; |
| const int probability = 10; |
| |
| const ProbabilityEntry entry(flag, probability); |
| const uint64_t encodedEntry = entry.encode(false /* hasHistoricalInfo */); |
| const ProbabilityEntry decodedEntry = |
| ProbabilityEntry::decode(encodedEntry, false /* hasHistoricalInfo */); |
| EXPECT_EQ(0xFF0Aull, encodedEntry); |
| EXPECT_EQ(flag, decodedEntry.getFlags()); |
| EXPECT_EQ(probability, decodedEntry.getProbability()); |
| } |
| |
| TEST(ProbabilityEntryTest, TestEncodeDecodeWithHistoricalInfo) { |
| const int flag = 0xF0; |
| const int timestamp = 0x3FFFFFFF; |
| const int level = 3; |
| const int count = 10; |
| |
| const HistoricalInfo historicalInfo(timestamp, level, count); |
| const ProbabilityEntry entry(flag, &historicalInfo); |
| |
| const uint64_t encodedEntry = entry.encode(true /* hasHistoricalInfo */); |
| EXPECT_EQ(0xF03FFFFFFF030Aull, encodedEntry); |
| const ProbabilityEntry decodedEntry = |
| ProbabilityEntry::decode(encodedEntry, true /* hasHistoricalInfo */); |
| |
| EXPECT_EQ(flag, decodedEntry.getFlags()); |
| EXPECT_EQ(timestamp, decodedEntry.getHistoricalInfo()->getTimestamp()); |
| EXPECT_EQ(level, decodedEntry.getHistoricalInfo()->getLevel()); |
| EXPECT_EQ(count, decodedEntry.getHistoricalInfo()->getCount()); |
| } |
| |
| } // namespace |
| } // namespace latinime |