<?php

declare(strict_types=1);

namespace tests\Preprocessing;

use Phpml\Preprocessing\Normalizer;

class NormalizerTest extends \PHPUnit_Framework_TestCase
{
    /**
     * @expectedException \Phpml\Exception\NormalizerException
     */
    public function testThrowExceptionOnInvalidNorm()
    {
        new Normalizer(99);
    }

    public function testNormalizeSamplesWithL2Norm()
    {
        $samples = [
            [1, -1, 2],
            [2, 0, 0],
            [0, 1, -1],
        ];

        $normalized = [
            [0.4, -0.4, 0.81],
            [1.0, 0.0, 0.0],
            [0.0, 0.7, -0.7],
        ];

        $normalizer = new Normalizer();
        $normalizer->transform($samples);

        $this->assertEquals($normalized, $samples, '', $delta = 0.01);
    }

    public function testNormalizeSamplesWithL1Norm()
    {
        $samples = [
            [1, -1, 2],
            [2, 0, 0],
            [0, 1, -1],
        ];

        $normalized = [
            [0.25, -0.25, 0.5],
            [1.0, 0.0, 0.0],
            [0.0, 0.5, -0.5],
        ];

        $normalizer = new Normalizer(Normalizer::NORM_L1);
        $normalizer->transform($samples);

        $this->assertEquals($normalized, $samples, '', $delta = 0.01);
    }

    public function testFitNotChangeNormalizerBehavior()
    {
        $samples = [
            [1, -1, 2],
            [2, 0, 0],
            [0, 1, -1],
        ];

        $normalized = [
            [0.4, -0.4, 0.81],
            [1.0, 0.0, 0.0],
            [0.0, 0.7, -0.7],
        ];

        $normalizer = new Normalizer();
        $normalizer->transform($samples);

        $this->assertEquals($normalized, $samples, '', $delta = 0.01);

        $normalizer->fit($samples);

        $this->assertEquals($normalized, $samples, '', $delta = 0.01);
    }

    public function testL1NormWithZeroSumCondition()
    {
        $samples = [
            [0, 0, 0],
            [2, 0, 0],
            [0, 1, -1],
        ];

        $normalized = [
            [0.33, 0.33, 0.33],
            [1.0, 0.0, 0.0],
            [0.0, 0.5, -0.5],
        ];

        $normalizer = new Normalizer(Normalizer::NORM_L1);
        $normalizer->transform($samples);

        $this->assertEquals($normalized, $samples, '', $delta = 0.01);
    }
}