endtoend-test-psl/tests/Psl/Math/MaxByTest.php

<?php

declare(strict_types=1);

namespace Psl\Tests\Math;

use Generator;
use PHPUnit\Framework\TestCase;
use Psl\Arr;
use Psl\Math;
use Psl\Str;

final class MaxByTest extends TestCase
{
    /**
     * @dataProvider provideData
     */
    public function testMaxBy($expected, array $values, callable $fun): void
    {
        static::assertSame($expected, Math\max_by($values, $fun));
    }

    public function provideData(): Generator
    {
        yield [
            'bazqux',
            ['foo', 'bar', 'baz', 'qux', 'foobar', 'bazqux'],
            static fn ($value) => Str\length($value),
        ];

        yield [
            ['foo', 'bar', 'baz'],
            [
                ['foo'],
                ['foo', 'bar'],
                ['foo', 'bar', 'baz']
            ],
            static fn ($arr) => Arr\count($arr),
        ];

        yield [
            9,
            [0, 1, 2, 3, 4, 5, 6, 7, 8, 9],
            static fn ($i) => $i,
        ];

        yield [
            null,
            [],
            static fn ($i) => $i,
        ];
    }
}