endtoend-test-psl/tests/Psl/Arr/TakeWhileTest.php

<?php

declare(strict_types=1);

namespace Psl\Tests\Arr;

use PHPUnit\Framework\TestCase;
use Psl\Arr;

class TakeWhileTest extends TestCase
{
    /**
     * @dataProvider provideData
     */
    public function testTakeWhile(array $expected, array $array, callable $callable): void
    {
        $result = Arr\take_while($array, $callable);

        self::assertSame($expected, $result);
    }

    public function provideData(): iterable
    {
        yield [[], [1, 2, 3, 4, 5], fn (int $_): bool => false];
        yield [[1, 2, 3], [1, 2, 3, 4, 5], fn (int $i) => $i <= 3];
        yield [[1, 2], [1, 2, 3, 4, 5], fn (int $i) => $i <= 2];
        yield [[1, 2, 3, 4, 5], [1, 2, 3, 4, 5], fn (int $_) => true];
    }
}