1//===- llvm/unittest/Support/EditDistanceTest.cpp - Edit distance tests ---===//
2//
3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6//
7//===----------------------------------------------------------------------===//
8
9#include "llvm/ADT/StringRef.h"
10#include "llvm/ADT/edit_distance.h"
11#include "gtest/gtest.h"
12#include <cstdlib>
13
14using namespace llvm;
15
16namespace {
17
18struct Result {
19 unsigned NumMaps;
20 unsigned EditDist;
21};
22} // namespace
23
24static Result editDistanceAndMaps(StringRef A, StringRef B,
25 unsigned MaxEditDistance = 0) {
26 unsigned NumMaps = 0;
27 auto TrackMaps = [&](const char X) {
28 ++NumMaps;
29 return X;
30 };
31 unsigned EditDist = llvm::ComputeMappedEditDistance(
32 FromArray: ArrayRef(A.data(), A.size()), ToArray: ArrayRef(B.data(), B.size()), Map: TrackMaps,
33 AllowReplacements: true, MaxEditDistance);
34 return {.NumMaps: NumMaps, .EditDist: EditDist};
35}
36
37TEST(EditDistance, VerifyShortCircuit) {
38 StringRef Hello = "Hello";
39 StringRef HelloWorld = "HelloWorld";
40 Result R = editDistanceAndMaps(A: Hello, B: HelloWorld, MaxEditDistance: 5);
41 EXPECT_EQ(R.EditDist, 5U);
42 EXPECT_GT(R.NumMaps, 0U);
43
44 R = editDistanceAndMaps(A: Hello, B: HelloWorld);
45 EXPECT_EQ(R.EditDist, 5U);
46 EXPECT_GT(R.NumMaps, 0U);
47
48 R = editDistanceAndMaps(A: Hello, B: HelloWorld, MaxEditDistance: 4);
49 EXPECT_EQ(R.EditDist, 5U);
50 EXPECT_EQ(R.NumMaps, 0U);
51
52 R = editDistanceAndMaps(A: HelloWorld, B: Hello, MaxEditDistance: 4);
53 EXPECT_EQ(R.EditDist, 5U);
54 EXPECT_EQ(R.NumMaps, 0U);
55
56 R = editDistanceAndMaps(A: Hello, B: HelloWorld, MaxEditDistance: 1);
57 EXPECT_EQ(R.EditDist, 2U);
58 EXPECT_EQ(R.NumMaps, 0U);
59
60 R = editDistanceAndMaps(A: HelloWorld, B: Hello, MaxEditDistance: 1);
61 EXPECT_EQ(R.EditDist, 2U);
62 EXPECT_EQ(R.NumMaps, 0U);
63}
64

source code of llvm/unittests/ADT/EditDistanceTest.cpp