<?php

declare (strict_types = 1);

namespace tests\Phpml\Dataset;

use Phpml\Dataset\FilesDataset;

class FilesDatasetTest extends \PHPUnit_Framework_TestCase
{
    /**
     * @expectedException \Phpml\Exception\DatasetException
     */
    public function testThrowExceptionOnMissingRootFolder()
    {
        new FilesDataset('some/not/existed/path');
    }

    public function testLoadFilesDatasetWithBBCData()
    {
        $rootPath = dirname(__FILE__).'/Resources/bbc';

        $dataset = new FilesDataset($rootPath);

        $this->assertEquals(50, count($dataset->getSamples()));
        $this->assertEquals(50, count($dataset->getTargets()));

        $targets = ['business', 'entertainment', 'politics', 'sport', 'tech'];
        $this->assertEquals($targets, array_values(array_unique($dataset->getTargets())));

        $firstSample = file_get_contents($rootPath.'/business/001.txt');
        $this->assertEquals($firstSample, $dataset->getSamples()[0][0]);

        $lastSample = file_get_contents($rootPath.'/tech/010.txt');
        $this->assertEquals($lastSample, $dataset->getSamples()[49][0]);
    }
    
}