php-ml/tests/Phpml/Dataset/ArrayDatasetTest.php
2017-02-03 12:58:25 +01:00

31 lines
686 B
PHP

<?php
declare(strict_types=1);
namespace tests\Phpml\Dataset;
use Phpml\Dataset\ArrayDataset;
use PHPUnit\Framework\TestCase;
class ArrayDatasetTest extends TestCase
{
/**
* @expectedException \Phpml\Exception\InvalidArgumentException
*/
public function testThrowExceptionOnInvalidArgumentsSize()
{
new ArrayDataset([0, 1], [0]);
}
public function testArrayDataset()
{
$dataset = new ArrayDataset(
$samples = [[1], [2], [3], [4]],
$labels = ['a', 'a', 'b', 'b']
);
$this->assertEquals($samples, $dataset->getSamples());
$this->assertEquals($labels, $dataset->getTargets());
}
}