Merge pull request #2248 from darkshredder/special-term-test

[nominatim.git] / test / php / Nominatim / PhraseTest.php

diff --git a/test/php/Nominatim/PhraseTest.php b/test/php/Nominatim/PhraseTest.php
index db8d8b50808ad8325dbcc769908d5fcbe0ee7d7b..42166e34f330da72a06f2089e6e99d0a4b73db21 100644 (file)
--- a/test/php/Nominatim/PhraseTest.php
+++ b/test/php/Nominatim/PhraseTest.php
@@ -2,10 +2,35 @@
 
 namespace Nominatim;
 
 
 namespace Nominatim;
 

-require_once '../../lib/Phrase.php';
+require_once(CONST_LibDir.'/Phrase.php');

 
 

-class PhraseTest extends \PHPUnit_Framework_TestCase
+class TokensFullSet

 {
 {

+    public function containsAny($sTerm)
+    {
+        return true;
+    }
+}
+
+// phpcs:ignore PSR1.Classes.ClassDeclaration.MultipleClasses
+class TokensPartialSet
+{
+    public function __construct($aTokens)
+    {
+        $this->aTokens = array_flip($aTokens);
+    }
+
+    public function containsAny($sTerm)
+    {
+        return isset($this->aTokens[$sTerm]);
+    }
+}
+
+// phpcs:ignore PSR1.Classes.ClassDeclaration.MultipleClasses
+class PhraseTest extends \PHPUnit\Framework\TestCase
+{
+
+

     private function serializeSets($aSets)
     {
         $aParts = array();
     private function serializeSets($aSets)
     {
         $aParts = array();


@@ -19,6 +44,7 @@ class PhraseTest extends \PHPUnit_Framework_TestCase
     public function testEmptyPhrase()
     {
         $oPhrase = new Phrase('', '');
     public function testEmptyPhrase()
     {
         $oPhrase = new Phrase('', '');

+        $oPhrase->computeWordSets(new TokensFullSet());

 
         $this->assertEquals(
             array(array('')),
 
         $this->assertEquals(
             array(array('')),


@@ -30,6 +56,7 @@ class PhraseTest extends \PHPUnit_Framework_TestCase
     public function testSingleWordPhrase()
     {
         $oPhrase = new Phrase('a', '');
     public function testSingleWordPhrase()
     {
         $oPhrase = new Phrase('a', '');

+        $oPhrase->computeWordSets(new TokensFullSet());

 
         $this->assertEquals(
             '(a)',
 
         $this->assertEquals(
             '(a)',


@@ -41,20 +68,23 @@ class PhraseTest extends \PHPUnit_Framework_TestCase
     public function testMultiWordPhrase()
     {
         $oPhrase = new Phrase('a b', '');
     public function testMultiWordPhrase()
     {
         $oPhrase = new Phrase('a b', '');

+        $oPhrase->computeWordSets(new TokensFullSet());

         $this->assertEquals(
             '(a b),(a|b)',
             $this->serializeSets($oPhrase->getWordSets())
         );
 
         $oPhrase = new Phrase('a b c', '');
         $this->assertEquals(
             '(a b),(a|b)',
             $this->serializeSets($oPhrase->getWordSets())
         );
 
         $oPhrase = new Phrase('a b c', '');

+        $oPhrase->computeWordSets(new TokensFullSet());

         $this->assertEquals(
         $this->assertEquals(

-            '(a b c),(a|b c),(a|b|c),(a b|c)',
+            '(a b c),(a|b c),(a b|c),(a|b|c)',

             $this->serializeSets($oPhrase->getWordSets())
         );
 
         $oPhrase = new Phrase('a b c d', '');
             $this->serializeSets($oPhrase->getWordSets())
         );
 
         $oPhrase = new Phrase('a b c d', '');

+        $oPhrase->computeWordSets(new TokensFullSet());

         $this->assertEquals(
         $this->assertEquals(

-            '(a b c d),(a|b c d),(a|b|c d),(a|b|c|d),(a|b c|d),(a b|c d),(a b|c|d),(a b c|d)',
+            '(a b c d),(a b c|d),(a b|c d),(a|b c d),(a b|c|d),(a|b c|d),(a|b|c d),(a|b|c|d)',

             $this->serializeSets($oPhrase->getWordSets())
         );
     }
             $this->serializeSets($oPhrase->getWordSets())
         );
     }


@@ -63,25 +93,47 @@ class PhraseTest extends \PHPUnit_Framework_TestCase
     public function testInverseWordSets()
     {
         $oPhrase = new Phrase('a b c', '');
     public function testInverseWordSets()
     {
         $oPhrase = new Phrase('a b c', '');

+        $oPhrase->computeWordSets(new TokensFullSet());

         $oPhrase->invertWordSets();
 
         $this->assertEquals(
         $oPhrase->invertWordSets();
 
         $this->assertEquals(

-            '(a b c),(c|a b),(c|b|a),(b c|a)',
+            '(a b c),(b c|a),(c|a b),(c|b|a)',

             $this->serializeSets($oPhrase->getWordSets())
         );
     }
 
 
             $this->serializeSets($oPhrase->getWordSets())
         );
     }
 
 

-    public function testMaxDepth()
+    public function testMaxWordSets()

     {
         $oPhrase = new Phrase(join(' ', array_fill(0, 4, 'a')), '');
     {
         $oPhrase = new Phrase(join(' ', array_fill(0, 4, 'a')), '');

+        $oPhrase->computeWordSets(new TokensFullSet());

         $this->assertEquals(8, count($oPhrase->getWordSets()));
         $oPhrase->invertWordSets();
         $this->assertEquals(8, count($oPhrase->getWordSets()));
 
         $oPhrase = new Phrase(join(' ', array_fill(0, 18, 'a')), '');
         $this->assertEquals(8, count($oPhrase->getWordSets()));
         $oPhrase->invertWordSets();
         $this->assertEquals(8, count($oPhrase->getWordSets()));
 
         $oPhrase = new Phrase(join(' ', array_fill(0, 18, 'a')), '');

-        $this->assertEquals(41226, count($oPhrase->getWordSets()));
+        $oPhrase->computeWordSets(new TokensFullSet());
+        $this->assertEquals(100, count($oPhrase->getWordSets()));

         $oPhrase->invertWordSets();
         $oPhrase->invertWordSets();

-        $this->assertEquals(41226, count($oPhrase->getWordSets()));
+        $this->assertEquals(100, count($oPhrase->getWordSets()));
+    }
+
+
+    public function testPartialTokensShortTerm()
+    {
+        $oPhrase = new Phrase('a b c d', '');
+        $oPhrase->computeWordSets(new TokensPartialSet(array('a', 'b', 'd', 'b c', 'b c d')));
+        $this->assertEquals(
+            '(a|b c d),(a|b c|d)',
+            $this->serializeSets($oPhrase->getWordSets())
+        );
+    }
+
+
+    public function testPartialTokensLongTerm()
+    {
+        $oPhrase = new Phrase(join(' ', array_fill(0, 18, 'a')), '');
+        $oPhrase->computeWordSets(new TokensPartialSet(array('a', 'a a a a a')));
+        $this->assertEquals(80, count($oPhrase->getWordSets()));

     }
 }
     }
 }