[search_tests] Fix bad m_resultType.

mapsme · Mar 11, 2021 · c284e16 · c284e16
1 parent 9ab88d2
commit c284e16
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/search/ranking_info.cpp b/search/ranking_info.cpp
@@ -258,7 +258,10 @@ double RankingInfo::GetLinearModelRank() const
     result += m_falseCats * kFalseCats;
     result += kType[m_type];
     if (Model::IsPoi(m_type))
+    {
+      CHECK_NOT_EQUAL(m_resultType, ResultType::Count, ());
       result += kResultType[base::Underlying(m_resultType)];
+    }
     result += (m_allTokensUsed ? 1 : 0) * kAllTokensUsed;
     result += (m_exactCountryOrCapital ? 1 : 0) * kExactCountryOrCapital;
     auto const nameRank = kNameScore[nameScore] + kErrorsMade * GetErrorsMadePerToken() +

diff --git a/search/search_tests/ranking_tests.cpp b/search/search_tests/ranking_tests.cpp
@@ -86,6 +86,7 @@ UNIT_TEST(PreferCountry)
   cafe.m_tokenRanges[Model::TYPE_SUBPOI] = TokenRange(0, 1);
   cafe.m_exactCountryOrCapital = false;
   cafe.m_type = Model::TYPE_SUBPOI;
+  cafe.m_resultType = ResultType::Eat;
 
   auto country = info;
   country.m_distanceToPivot = 1e6;