41 lines
1.0 KiB
PHP
41 lines
1.0 KiB
PHP
<?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());
|
|
}
|
|
}
|