niceSOFT/python3-charset-normalizer: Truly universal encoding detector in pure Python.

tests/test_isbinary.py

29 строк736 B

TAHRI Ahmed R.

:art: refresh/upgrade project structure (#558)

29 окт 2024, 10:11

29 окт 2024, 10:11a34a103

from __future__ import annotations

import typing
from base64 import b64decode
from io import BytesIO
from os import pardir, path

import pytest

from charset_normalizer import is_binary

DIR_PATH = path.join(path.dirname(path.realpath(__file__)), pardir)


@pytest.mark.parametrize(
    "raw, expected",
    [
        (b"\x00\x5f\x2f\xff" * 50, True),
        (b64decode("R0lGODlhAQABAAAAACw="), True),
        (BytesIO(b64decode("R0lGODlhAQABAAAAACw=")), True),
        ("sample-polish.txt", False),
        ("sample-arabic.txt", False),
    ],
)
def test_isbinary(raw: bytes | typing.BinaryIO | str, expected: bool) -> None:
    if isinstance(raw, str):
        raw = DIR_PATH + f"/data/{raw}"

    assert is_binary(raw) is expected