php-ml/tests/Phpml/ModelManagerTest.php

<?php

declare(strict_types=1);

namespace tests;

use Phpml\ModelManager;
use Phpml\Regression\LeastSquares;
use PHPUnit\Framework\TestCase;

class ModelManagerTest extends TestCase
{
    public function testSaveAndRestore()
    {
        $filename = 'test-save-to-file-'.rand(100, 999).'-'.uniqid();
        $filepath = tempnam(sys_get_temp_dir(), $filename);

        $obj = new LeastSquares();
        $modelManager = new ModelManager();
        $modelManager->saveToFile($obj, $filepath);

        $restored = $modelManager->restoreFromFile($filepath);
        $this->assertEquals($obj, $restored);
    }

    /**
     * @expectedException \Phpml\Exception\FileException
     */
    public function testSaveToWrongFile()
    {
        $filepath = sys_get_temp_dir() . DIRECTORY_SEPARATOR . 'unexisting';

        $obj = new LeastSquares();
        $modelManager = new ModelManager();
        $modelManager->saveToFile($obj, $filepath);
    }

    /**
     * @expectedException \Phpml\Exception\FileException
     */
    public function testRestoreWrongFile()
    {
        $filepath = sys_get_temp_dir() . DIRECTORY_SEPARATOR . 'unexisting';
        $modelManager = new ModelManager();
        $modelManager->restoreFromFile($filepath);
    }
}