1 files changed, 55 insertions, 0 deletions
diff --git a/core/source/benchmark/scala/com/rockymadden/stringmetric/similarity/WeightedLevenshteinMetricBenchmark.scala b/core/source/benchmark/scala/com/rockymadden/stringmetric/similarity/WeightedLevenshteinMetricBenchmark.scala
new file mode 100755
index 0000000..b511654
--- /dev/null
+++ b/core/source/benchmark/scala/com/rockymadden/stringmetric/similarity/WeightedLevenshteinMetricBenchmark.scala
@@ -0,0 +1,55 @@
+package com.rockymadden.stringmetric.similarity
+
+import com.google.caliper.Param
+import com.rockymadden.stringmetric.{CaliperBenchmark, CaliperRunner}
+import scala.annotation.tailrec
+import scala.util.Random
+
+final class WeightedLevenshteinMetricBenchmark extends CaliperBenchmark {
+	import WeightedLevenshteinMetricBenchmark.Metric
+
+	@Param(Array("0", "1", "2", "4", "8", "16"))
+	var length: Int = _
+
+	var string1: String = _
+	var charArray1: Array[Char] = _
+	var string2: String = _
+	var charArray2: Array[Char] = _
+
+	override protected def setUp() {
+		@tailrec
+		def random(l: Int, ps: String = null): String =
+			if (l == 0) ""
+			else {
+				val s = Random.alphanumeric.take(l).mkString
+
+				if (ps == null || s != ps) s
+				else random(l, ps)
+			}
+
+		string1 = random(length)
+		string2 = random(length, string1)
+		charArray1 = string1.toCharArray
+		charArray2 = string2.toCharArray
+	}
+
+	def timeCompareWithDifferentCharArrays(reps: Int) = run(reps) {
+		Metric.compare(charArray1, charArray2)(1, 1, 1)
+	}
+
+	def timeCompareWithDifferentStrings(reps: Int) = run(reps) {
+		Metric.compare(string1, string2)(1, 1, 1)
+	}
+
+	def timeCompareWithIdenticalCharArrays(reps: Int) = run(reps) {
+		Metric.compare(charArray1, charArray1)(1, 1, 1)
+	}
+
+	def timeCompareWithIdenticalStrings(reps: Int) = run(reps) {
+		Metric.compare(string1, string1)(1, 1, 1)
+	}
+}
+
+object WeightedLevenshteinMetricBenchmark extends CaliperRunner(classOf[WeightedLevenshteinMetricBenchmark]) {
+	private final val Metric = WeightedLevenshteinMetric()
+}