34 lines
852 B
PHP
34 lines
852 B
PHP
<?php
|
|
|
|
declare(strict_types=1);
|
|
|
|
namespace Phpml\Tests\Dataset;
|
|
|
|
use Phpml\Dataset\MnistDataset;
|
|
use Phpml\Exception\InvalidArgumentException;
|
|
use PHPUnit\Framework\TestCase;
|
|
|
|
class MnistDatasetTest extends TestCase
|
|
{
|
|
public function testSimpleMnistDataset(): void
|
|
{
|
|
$dataset = new MnistDataset(
|
|
__DIR__.'/Resources/mnist/images-idx-ubyte',
|
|
__DIR__.'/Resources/mnist/labels-idx-ubyte'
|
|
);
|
|
|
|
self::assertCount(10, $dataset->getSamples());
|
|
self::assertCount(10, $dataset->getTargets());
|
|
}
|
|
|
|
public function testCheckSamplesAndTargetsCountMatch(): void
|
|
{
|
|
$this->expectException(InvalidArgumentException::class);
|
|
|
|
new MnistDataset(
|
|
__DIR__.'/Resources/mnist/images-idx-ubyte',
|
|
__DIR__.'/Resources/mnist/labels-11-idx-ubyte'
|
|
);
|
|
}
|
|
}
|