summaryrefslogtreecommitdiff
path: root/core/src/test/scala/com/rockymadden/stringmetric/similarity/NGramMetricSpec.scala
blob: 501a331b623b981d6725fbecca3cf798a3432e04 (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
32
33
34
35
36
37
38
39
40
41
42
43
44
45
package com.rockymadden.stringmetric.similarity

object NGramMetricSpec extends org.specs2.mutable.SpecificationWithJUnit {
	"NGramMetric compare()" should {
		"return None with empty arguments" in {
			NGramMetric(1).compare("", "").isDefined must beFalse
			NGramMetric(1).compare("abc", "").isDefined must beFalse
			NGramMetric(1).compare("", "xyz").isDefined must beFalse
		}
		"return 1 with equal arguments" in {
			NGramMetric(1).compare("abc", "abc").get must beEqualTo(1)
			NGramMetric(2).compare("abc", "abc").get must beEqualTo(1)
			NGramMetric(3).compare("abc", "abc").get must beEqualTo(1)
		}
		"return 0 with unequal arguments" in {
			NGramMetric(1).compare("abc", "xyz").get must beEqualTo(0)
			NGramMetric(2).compare("abc", "xyz").get must beEqualTo(0)
			NGramMetric(3).compare("abc", "xyz").get must beEqualTo(0)
		}
		"return None with invalid arguments" in {
			NGramMetric(2).compare("n", "naght").isDefined must beFalse
			NGramMetric(2).compare("night", "n").isDefined must beFalse
			NGramMetric(3).compare("ni", "naght").isDefined must beFalse
			NGramMetric(3).compare("night", "na").isDefined must beFalse
		}
		"return distance with valid arguments" in {
			NGramMetric(1).compare("night", "nacht").get must beEqualTo(0.6)
			NGramMetric(1).compare("night", "naght").get must beEqualTo(0.8)
			NGramMetric(1).compare("context", "contact").get must beEqualTo(0.7142857142857143)

			NGramMetric(2).compare("night", "nacht").get must beEqualTo(0.25)
			NGramMetric(2).compare("night", "naght").get must beEqualTo(0.5)
			NGramMetric(2).compare("context", "contact").get must beEqualTo(0.5)
			NGramMetric(2).compare("contextcontext", "contact").get must beEqualTo(0.23076923076923078)
			NGramMetric(2).compare("context", "contactcontact").get must beEqualTo(0.23076923076923078)
			NGramMetric(2).compare("ht", "nacht").get must beEqualTo(0.25)
			NGramMetric(2).compare("xp", "nacht").get must beEqualTo(0)
			NGramMetric(2).compare("ht", "hththt").get must beEqualTo(0.2)

			NGramMetric(3).compare("night", "nacht").get must beEqualTo(0)
			NGramMetric(3).compare("night", "naght").get must beEqualTo(0.3333333333333333)
			NGramMetric(3).compare("context", "contact").get must beEqualTo(0.4)
		}
	}
}