Current File : /home/inlingua/miniconda3/pkgs/rich-13.9.4-py312h06a4308_0/info/test/tests/test_getfileno.py

from rich._fileno import get_fileno


def test_get_fileno():
    class FileLike:
        def fileno(self) -> int:
            return 123

    assert get_fileno(FileLike()) == 123


def test_get_fileno_missing():
    class FileLike:
        pass

    assert get_fileno(FileLike()) is None


def test_get_fileno_broken():
    class FileLike:
        def fileno(self) -> int:
            1 / 0
            return 123

    assert get_fileno(FileLike()) is None