summaryrefslogtreecommitdiff
path: root/core/source/benchmark/scala/org/hashtree/stringmetric/similarity/NGramAlgorithmBenchmark.scala
blob: 9bef2fa696d14f08348e3087130a2425dbe0551b (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
package org.hashtree.stringmetric.similarity

import com.google.caliper.Param
import org.hashtree.stringmetric.{ CaliperBenchmark, CaliperRunner }
import scala.util.Random

final class NGramAlgorithmBenchmark extends CaliperBenchmark {
	@Param(Array("0", "1", "2", "4", "8", "16"))
	var length: Int = _

	@Param(Array("2", "3"))
	var n: Int = _

	var string: String = _
	var charArray: Array[Char] = _

	override protected def setUp() {
		string = Random.alphanumeric.take(length).mkString
		charArray = string.toCharArray
	}

	def timeComputeWithCharArray(reps: Int) = run(reps) {
		NGramAlgorithm.compute(charArray)(n)
	}

	def timeComputeWithString(reps: Int) = run(reps) {
		NGramAlgorithm.compute(string)(n)
	}
}

object NGramAlgorithmBenchmark extends CaliperRunner(classOf[NGramAlgorithmBenchmark])