1 /*
2 * Copyright (C) 2014 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17 #include "dictionary/structure/v4/content/probability_entry.h"
18
19 #include <gtest/gtest.h>
20
21 #include "defines.h"
22
23 namespace latinime {
24 namespace {
25
TEST(ProbabilityEntryTest,TestEncodeDecode)26 TEST(ProbabilityEntryTest, TestEncodeDecode) {
27 const int flag = 0xFF;
28 const int probability = 10;
29
30 const ProbabilityEntry entry(flag, probability);
31 const uint64_t encodedEntry = entry.encode(false /* hasHistoricalInfo */);
32 const ProbabilityEntry decodedEntry =
33 ProbabilityEntry::decode(encodedEntry, false /* hasHistoricalInfo */);
34 EXPECT_EQ(0xFF0Aull, encodedEntry);
35 EXPECT_EQ(flag, decodedEntry.getFlags());
36 EXPECT_EQ(probability, decodedEntry.getProbability());
37 }
38
TEST(ProbabilityEntryTest,TestEncodeDecodeWithHistoricalInfo)39 TEST(ProbabilityEntryTest, TestEncodeDecodeWithHistoricalInfo) {
40 const int flag = 0xF0;
41 const int timestamp = 0x3FFFFFFF;
42 const int count = 0xABCD;
43
44 const HistoricalInfo historicalInfo(timestamp, 0 /* level */, count);
45 const ProbabilityEntry entry(flag, &historicalInfo);
46
47 const uint64_t encodedEntry = entry.encode(true /* hasHistoricalInfo */);
48 EXPECT_EQ(0xF03FFFFFFFABCDull, encodedEntry);
49 const ProbabilityEntry decodedEntry =
50 ProbabilityEntry::decode(encodedEntry, true /* hasHistoricalInfo */);
51
52 EXPECT_EQ(flag, decodedEntry.getFlags());
53 EXPECT_EQ(timestamp, decodedEntry.getHistoricalInfo()->getTimestamp());
54 EXPECT_EQ(count, decodedEntry.getHistoricalInfo()->getCount());
55 }
56
57 } // namespace
58 } // namespace latinime
59