<?php

declare(strict_types=1);

namespace tests\Phpml\FeatureExtraction;

use Phpml\FeatureExtraction\TfIdfTransformer;
use PHPUnit\Framework\TestCase;

class TfIdfTransformerTest extends TestCase
{
    public function testTfIdfTransformation(): void
    {
        // https://en.wikipedia.org/wiki/Tf-idf

        $samples = [
            [0 => 1, 1 => 1, 2 => 2, 3 => 1, 4 => 0, 5 => 0],
            [0 => 1, 1 => 1, 2 => 0, 3 => 0, 4 => 2, 5 => 3],
        ];

        $tfIdfSamples = [
            [0 => 0, 1 => 0, 2 => 0.602, 3 => 0.301, 4 => 0, 5 => 0],
            [0 => 0, 1 => 0, 2 => 0, 3 => 0, 4 => 0.602, 5 => 0.903],
        ];

        $transformer = new TfIdfTransformer($samples);
        $transformer->transform($samples);

        $this->assertEquals($tfIdfSamples, $samples, '', 0.001);
    }
}