leetcode/string_match/length_of_last_word.py

import pytest

def lengthOfLastWord(s: str) -> int:
    s = s.split()[-1]
    return len(s)

@pytest.mark.parametrize(
    "s, expect",
    [
        ("Hello World", 5),
        ("   fly me   to   the moon  ", 4),
        ("luffy is still joyboy", 6)
    ]
)

def test_cases(s, expect):
    assert lengthOfLastWord(s) == expect