php-ml/tests/Phpml/Dataset/ArrayDatasetTest.php
2016-06-16 23:56:15 +02:00

30 lines
676 B
PHP

<?php
declare (strict_types = 1);
namespace tests\Phpml\Dataset;
use Phpml\Dataset\ArrayDataset;
class ArrayDatasetTest extends \PHPUnit_Framework_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());
}
}