endtoend-test-psl/tests/Psl/Iter/TakeTest.php
2020-10-15 11:05:30 +02:00

28 lines
634 B
PHP

<?php
declare(strict_types=1);
namespace Psl\Tests\Iter;
use PHPUnit\Framework\TestCase;
use Psl;
use Psl\Iter;
final class TakeTest extends TestCase
{
public function testTake(): void
{
$result = Iter\take(Iter\to_iterator([-5, -4, -3, -2, -1, 0, 1, 2, 3, 4, 5]), 3);
static::assertSame([-5, -4, -3], Iter\to_array($result));
}
public function testTakeThrowsIfLengthIsNegative(): void
{
$this->expectException(Psl\Exception\InvariantViolationException::class);
$this->expectExceptionMessage('Length must be non-negative.');
Iter\take([1, 2, 3], -3);
}
}