58 lines
2 KiB
PHP
58 lines
2 KiB
PHP
|
<?php
|
||
|
|
namespace Tests\TextAnalysis\Collections;
|
||
|
|
use TextAnalysis\Collections\DocumentArrayCollection;
|
||
|
|
use TextAnalysis\Documents\TokensDocument;
|
||
|
|
|
||
|
|
use TextAnalysis\Filters\QuotesFilter;
|
||
|
|
use TextAnalysis\Filters\LowerCaseFilter;
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Test building an index works with collections
|
||
|
|
*
|
||
|
|
* @author yooper
|
||
|
|
*/
|
||
|
|
class CollectionInvertedBuilderTest extends \PHPUnit\Framework\TestCase
|
||
|
|
{
|
||
|
|
public function testSimpleCollectionIndex()
|
||
|
|
{
|
||
|
|
$docs = array(
|
||
|
|
new TokensDocument(array("marquette", "michigan", "hiking", "hiking", "hiking" , "camping", "swimming")),
|
||
|
|
new TokensDocument(array("ironwood", "michigan", "hiking", "biking", "camping", "swimming","marquette")),
|
||
|
|
new TokensDocument(array("no","tokens"))
|
||
|
|
);
|
||
|
|
|
||
|
|
$collection = new DocumentArrayCollection($docs);
|
||
|
|
|
||
|
|
$this->assertTrue($collection->count() === 3);
|
||
|
|
|
||
|
|
$this->assertEquals($collection[2]->getDocumentData(), array("no","tokens"));
|
||
|
|
}
|
||
|
|
|
||
|
|
public function testFiltersOnCollection()
|
||
|
|
{
|
||
|
|
$docs = array(
|
||
|
|
new TokensDocument(array("Marquette", "Michigan's", "hiking", "hiking", "hiking" , "camping", "swimming")),
|
||
|
|
new TokensDocument(array("Ironwood", "michigan", "hiking", "biking", "camping", "swimming","marquette")),
|
||
|
|
new TokensDocument(array("No","Tokens"))
|
||
|
|
);
|
||
|
|
|
||
|
|
$collection = new DocumentArrayCollection($docs);
|
||
|
|
|
||
|
|
$filters = array(
|
||
|
|
new LowerCaseFilter(),
|
||
|
|
new QuotesFilter()
|
||
|
|
);
|
||
|
|
|
||
|
|
$collection->applyTransformations($filters);
|
||
|
|
|
||
|
|
|
||
|
|
$this->assertTrue($collection->count() === 3);
|
||
|
|
|
||
|
|
$this->assertEquals(array("marquette", "michigans", "hiking", "hiking", "hiking" , "camping", "swimming"), $collection[0]->getDocumentData());
|
||
|
|
$this->assertEquals(array("ironwood", "michigan", "hiking", "biking", "camping", "swimming","marquette"),$collection[1]->getDocumentData());
|
||
|
|
$this->assertEquals(array("no","tokens"), $collection[2]->getDocumentData());
|
||
|
|
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|