관리-도구

편집 파일: rec.pyi

import io
from typing import Any

import numpy as np
import numpy.typing as npt

AR_i8: npt.NDArray[np.int64]
REC_AR_V: np.recarray[Any, np.dtype[np.record]]
AR_LIST: list[npt.NDArray[np.int64]]

format_parser: np.format_parser
record: np.record
file_obj: io.BufferedIOBase

reveal_type(np.format_parser(  # E: format_parser
    formats=[np.float64, np.int64, np.bool_],
    names=["f8", "i8", "?"],
    titles=None,
    aligned=True,
))
reveal_type(format_parser.dtype)  # E: dtype[void]

reveal_type(record.field_a)  # E: Any
reveal_type(record.field_b)  # E: Any
reveal_type(record["field_a"])  # E: Any
reveal_type(record["field_b"])  # E: Any
reveal_type(record.pprint())  # E: str
record.field_c = 5

reveal_type(REC_AR_V.field(0))  # E: Any
reveal_type(REC_AR_V.field("field_a"))  # E: Any
reveal_type(REC_AR_V.field(0, AR_i8))  # E: None
reveal_type(REC_AR_V.field("field_a", AR_i8))  # E: None
reveal_type(REC_AR_V["field_a"])  # E: Any
reveal_type(REC_AR_V.field_a)  # E: Any
reveal_type(REC_AR_V.__array_finalize__(object()))  # E: None

reveal_type(np.recarray(  # recarray[Any, dtype[record]]
    shape=(10, 5),
    formats=[np.float64, np.int64, np.bool_],
    order="K",
    byteorder="|",
))
reveal_type(np.recarray(  # recarray[Any, dtype[Any]]
    shape=(10, 5),
    dtype=[("f8", np.float64), ("i8", np.int64)],
    strides=(5, 5),
))

reveal_type(np.rec.fromarrays(  # recarray[Any, dtype[record]]
    AR_LIST,
))
reveal_type(np.rec.fromarrays(  # recarray[Any, dtype[Any]]
    AR_LIST,
    dtype=np.int64,
))
reveal_type(np.rec.fromarrays(  # recarray[Any, dtype[Any]]
    AR_LIST,
    formats=[np.int64, np.float64],
    names=["i8", "f8"]
))

reveal_type(np.rec.fromrecords(  # recarray[Any, dtype[record]]
    (1, 1.5),
))
reveal_type(np.rec.fromrecords(  # recarray[Any, dtype[record]]
    [(1, 1.5)],
    dtype=[("i8", np.int64), ("f8", np.float64)],
))
reveal_type(np.rec.fromrecords(  # recarray[Any, dtype[record]]
    REC_AR_V,
    formats=[np.int64, np.float64],
    names=["i8", "f8"]
))

reveal_type(np.rec.fromstring(  # recarray[Any, dtype[record]]
    b"(1, 1.5)",
    dtype=[("i8", np.int64), ("f8", np.float64)],
))
reveal_type(np.rec.fromstring(  # recarray[Any, dtype[record]]
    REC_AR_V,
    formats=[np.int64, np.float64],
    names=["i8", "f8"]
))

reveal_type(np.rec.fromfile(  # recarray[Any, dtype[Any]]
    "test_file.txt",
    dtype=[("i8", np.int64), ("f8", np.float64)],
))
reveal_type(np.rec.fromfile(  # recarray[Any, dtype[record]]
    file_obj,
    formats=[np.int64, np.float64],
    names=["i8", "f8"]
))

reveal_type(np.rec.array(  # recarray[Any, dtype[{int64}]]
    AR_i8,
))
reveal_type(np.rec.array(  # recarray[Any, dtype[Any]]
    [(1, 1.5)],
    dtype=[("i8", np.int64), ("f8", np.float64)],
))
reveal_type(np.rec.array(  # recarray[Any, dtype[record]]
    [(1, 1.5)],
    formats=[np.int64, np.float64],
    names=["i8", "f8"]
))

reveal_type(np.rec.array(  # recarray[Any, dtype[Any]]
    None,
    dtype=np.float64,
    shape=(10, 3),
))
reveal_type(np.rec.array(  # recarray[Any, dtype[Any]]
    None,
    formats=[np.int64, np.float64],
    names=["i8", "f8"],
    shape=(10, 3),
))
reveal_type(np.rec.array(  # recarray[Any, dtype[Any]]
    file_obj,
    dtype=np.float64,
))
reveal_type(np.rec.array(  # recarray[Any, dtype[Any]]
    file_obj,
    formats=[np.int64, np.float64],
    names=["i8", "f8"],
))