endtoend-test-psl/tests/Psl/Iter/TakeWhileTest.php
2020-09-01 06:50:23 +01:00

40 lines
1.3 KiB
PHP

<?php
declare(strict_types=1);
namespace Psl\Tests\Iter;
use PHPUnit\Framework\TestCase;
use Psl\Iter;
class TakeWhileTest extends TestCase
{
/**
* @dataProvider provideData
*/
public function testTakeWhile(array $expected, iterable $iterable, callable $callable): void
{
$result = Iter\take_while($iterable, $callable);
self::assertSame($expected, Iter\to_array_with_keys($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];
yield [[], Iter\range(1, 5), fn (int $_): bool => false];
yield [[1, 2, 3], Iter\range(1, 5), fn (int $i) => $i <= 3];
yield [[1, 2], Iter\range(1, 5), fn (int $i) => $i <= 2];
yield [[1, 2, 3, 4, 5], Iter\range(1, 5), fn (int $_) => true];
yield [[], Iter\range(1, 5), fn (int $_): bool => false];
yield [[1, 2 , 3], Iter\range(1, 5), fn (int $i) => $i <= 3];
yield [[1, 2], Iter\range(1, 5), fn (int $i) => $i <= 2];
yield [[1, 2, 3, 4, 5], Iter\range(1, 5), fn (int $_) => true];
}
}