<?php

declare(strict_types=1);

namespace Phpml\Tests\Dataset;

use Phpml\Dataset\ArrayDataset;
use Phpml\Exception\InvalidArgumentException;
use PHPUnit\Framework\TestCase;

class ArrayDatasetTest extends TestCase
{
    public function testThrowExceptionOnInvalidArgumentsSize(): void
    {
        $this->expectException(InvalidArgumentException::class);
        new ArrayDataset([0, 1], [0]);
    }

    public function testArrayDataset(): void
    {
        $dataset = new ArrayDataset(
            $samples = [[1], [2], [3], [4]],
            $labels = ['a', 'a', 'b', 'b']
        );

        self::assertEquals($samples, $dataset->getSamples());
        self::assertEquals($labels, $dataset->getTargets());
    }

    public function testRemoveColumns(): void
    {
        $dataset = new ArrayDataset(
            [[1, 2, 3, 4], [2, 3, 4, 5], [3, 4, 5, 6], [4, 5, 6, 7]],
            ['a', 'a', 'b', 'b']
        );
        $dataset->removeColumns([0, 2]);

        self::assertEquals([[2, 4], [3, 5], [4, 6], [5, 7]], $dataset->getSamples());
    }
}