search/search_integration_tests/utils_test.cpp - master - githubmirror/omim

search/search_integration_tests/utils_test.cpp
115 строк4 KB
tatiana-yan
[search] Add IsCategorialRequestFuzzy method.
19 мар 2019, 18:31
19 мар 2019, 18:3185faa2d
#include "testing/testing.hpp"

#include "search/search_tests_support/helpers.hpp"

#include "generator/feature_builder.hpp"
#include "generator/generator_tests_support/test_feature.hpp"
#include "generator/generator_tests_support/test_mwm_builder.hpp"

#include "indexer/categories_holder.hpp"
#include "indexer/classificator.hpp"
#include "indexer/feature_decl.hpp"

#include "geometry/point2d.hpp"
#include "geometry/rect2d.hpp"

#include <algorithm>
#include <string>
#include <vector>

using namespace generator::tests_support;
using namespace search::tests_support;
using namespace std;

namespace search
{
namespace
{
class SearchUtilsTest : public SearchTest
{
public:
  DataSource const & GetDataSource() const { return m_dataSource; }
};

UNIT_CLASS_TEST(SearchUtilsTest, Utils)
{
  string const kCountryName = "FoodLand";
  auto file = platform::LocalCountryFile::MakeForTesting(kCountryName);

  TestPOI cafe(m2::PointD(0.0, 0.0), "cafe", "en");
  cafe.SetTypes({{"amenity", "cafe"}});

  TestPOI restaurant(m2::PointD(0.0, 0.0), "restaurant", "en");
  restaurant.SetTypes({{"amenity", "restaurant"}});

  TestPOI bar(m2::PointD(0.0, 0.0), "bar", "en");
  bar.SetTypes({{"amenity", "bar"}});

  auto id = BuildCountry(kCountryName, [&](TestMwmBuilder & builder) {
    builder.Add(cafe);
    builder.Add(restaurant);
    builder.Add(bar);
  });

  auto const & categories = GetDefaultCategories();
  auto const typesCafe = GetCategoryTypes("Cafe", "en", categories);
  auto const typesRestaurant = GetCategoryTypes("Restaurant", "en", categories);
  auto const typesBar = GetCategoryTypes("Bar", "en", categories);
  auto const typesFood = GetCategoryTypes("Eat", "en", categories);

  // GetCategoryTypes must return sorted vector of types.
  TEST(is_sorted(typesCafe.begin(), typesCafe.end()), ());
  TEST(is_sorted(typesRestaurant.begin(), typesRestaurant.end()), ());
  TEST(is_sorted(typesBar.begin(), typesBar.end()), ());
  TEST(is_sorted(typesFood.begin(), typesFood.end()), ());

  // Now "Cafe" and "Restaurant" are synonyms in categories.
  TEST_EQUAL(typesCafe, typesRestaurant, ());
  TEST_NOT_EQUAL(typesCafe, typesBar, ());

  for (auto const t : typesCafe)
    TEST(binary_search(typesFood.begin(), typesFood.end(), t), ());

  for (auto const t : typesBar)
    TEST(binary_search(typesFood.begin(), typesFood.end(), t), ());

  auto const & dataSource = GetDataSource();
  auto const rect = m2::RectD(m2::PointD(-0.5, -0.5), m2::PointD(0.5, 0.5));

  auto const testTypes = [&](vector<uint32_t> const & types, size_t expectedCount) {
    vector<FeatureID> features;
    ForEachOfTypesInRect(dataSource, types, rect,
                         [&features](FeatureID const & f) { features.push_back(f); });
    TEST_EQUAL(features.size(), expectedCount, ());
  };

  // |cafe| and |restaurant|.
  testTypes(typesCafe, 2);

  // |cafe| and |restaurant|.
  testTypes(typesRestaurant, 2);

  // |bar|.
  testTypes(typesBar, 1);

  // All.
  testTypes(typesFood, 3);
}

UNIT_TEST(IsCategorialRequestFuzzy)
{
  auto const isHotelRequest = [](string const & q) { return IsCategorialRequestFuzzy(q, "hotel"); };

  TEST(isHotelRequest("hotel"), ());
  TEST(isHotelRequest("Hotel"), ());
  TEST(isHotelRequest("motel"), ());
  TEST(isHotelRequest("отель"), ());
  TEST(isHotelRequest("гостиница"), ());
  TEST(isHotelRequest("гостиница москва"), ());
  TEST(isHotelRequest("new york hotel"), ());
  TEST(!isHotelRequest("new york where to stay"), ());
  TEST(!isHotelRequest("where to stay"), ());
  TEST(!isHotelRequest("random request"), ());
}
}  // namespace
}  // namespace search