Datasetpublic
codekingpro/portable-devtools

sourceHugging Faceupdated 5mo agoView on Hugging Face
1likes14kdownloads
rootpythonuser_packagesPython313site-packagesnumpytypingtestsdatarevealarraysetops.pyi
arraysetops.pyi75 linesDownload Raw Back to reveal
1from typing import Any, assert_type
2
3import numpy as np
4import numpy.typing as npt
5from numpy.lib._arraysetops_impl import (
6    UniqueAllResult,
7    UniqueCountsResult,
8    UniqueInverseResult,
9)
10
11AR_b: npt.NDArray[np.bool]
12AR_i8: npt.NDArray[np.int64]
13AR_f8: npt.NDArray[np.float64]
14AR_M: npt.NDArray[np.datetime64]
15AR_O: npt.NDArray[np.object_]
16
17AR_LIKE_f8: list[float]
18
19assert_type(np.ediff1d(AR_b), npt.NDArray[np.int8])
20assert_type(np.ediff1d(AR_i8, to_end=[1, 2, 3]), npt.NDArray[np.int64])
21assert_type(np.ediff1d(AR_M), npt.NDArray[np.timedelta64])
22assert_type(np.ediff1d(AR_O), npt.NDArray[np.object_])
23assert_type(np.ediff1d(AR_LIKE_f8, to_begin=[1, 1.5]), npt.NDArray[Any])
24
25assert_type(np.intersect1d(AR_i8, AR_i8), npt.NDArray[np.int64])
26assert_type(np.intersect1d(AR_M, AR_M, assume_unique=True), npt.NDArray[np.datetime64])
27assert_type(np.intersect1d(AR_f8, AR_i8), npt.NDArray[Any])
28assert_type(
29    np.intersect1d(AR_f8, AR_f8, return_indices=True),
30    tuple[npt.NDArray[np.float64], npt.NDArray[np.intp], npt.NDArray[np.intp]],
31)
32
33assert_type(np.setxor1d(AR_i8, AR_i8), npt.NDArray[np.int64])
34assert_type(np.setxor1d(AR_M, AR_M, assume_unique=True), npt.NDArray[np.datetime64])
35assert_type(np.setxor1d(AR_f8, AR_i8), npt.NDArray[Any])
36
37assert_type(np.isin(AR_i8, AR_i8), npt.NDArray[np.bool])
38assert_type(np.isin(AR_M, AR_M, assume_unique=True), npt.NDArray[np.bool])
39assert_type(np.isin(AR_f8, AR_i8), npt.NDArray[np.bool])
40assert_type(np.isin(AR_f8, AR_LIKE_f8, invert=True), npt.NDArray[np.bool])
41
42assert_type(np.union1d(AR_i8, AR_i8), npt.NDArray[np.int64])
43assert_type(np.union1d(AR_M, AR_M), npt.NDArray[np.datetime64])
44assert_type(np.union1d(AR_f8, AR_i8), npt.NDArray[Any])
45
46assert_type(np.setdiff1d(AR_i8, AR_i8), npt.NDArray[np.int64])
47assert_type(np.setdiff1d(AR_M, AR_M, assume_unique=True), npt.NDArray[np.datetime64])
48assert_type(np.setdiff1d(AR_f8, AR_i8), npt.NDArray[Any])
49
50assert_type(np.unique(AR_f8), npt.NDArray[np.float64])
51assert_type(np.unique(AR_LIKE_f8, axis=0), npt.NDArray[Any])
52assert_type(np.unique(AR_f8, return_index=True), tuple[npt.NDArray[np.float64], npt.NDArray[np.intp]])
53assert_type(np.unique(AR_LIKE_f8, return_index=True), tuple[npt.NDArray[Any], npt.NDArray[np.intp]])
54assert_type(np.unique(AR_f8, return_inverse=True), tuple[npt.NDArray[np.float64], npt.NDArray[np.intp]])
55assert_type(np.unique(AR_LIKE_f8, return_inverse=True), tuple[npt.NDArray[Any], npt.NDArray[np.intp]])
56assert_type(np.unique(AR_f8, return_counts=True), tuple[npt.NDArray[np.float64], npt.NDArray[np.intp]])
57assert_type(np.unique(AR_LIKE_f8, return_counts=True), tuple[npt.NDArray[Any], npt.NDArray[np.intp]])
58assert_type(np.unique(AR_f8, return_index=True, return_inverse=True), tuple[npt.NDArray[np.float64], npt.NDArray[np.intp], npt.NDArray[np.intp]])
59assert_type(np.unique(AR_LIKE_f8, return_index=True, return_inverse=True), tuple[npt.NDArray[Any], npt.NDArray[np.intp], npt.NDArray[np.intp]])
60assert_type(np.unique(AR_f8, return_index=True, return_counts=True), tuple[npt.NDArray[np.float64], npt.NDArray[np.intp], npt.NDArray[np.intp]])
61assert_type(np.unique(AR_LIKE_f8, return_index=True, return_counts=True), tuple[npt.NDArray[Any], npt.NDArray[np.intp], npt.NDArray[np.intp]])
62assert_type(np.unique(AR_f8, return_inverse=True, return_counts=True), tuple[npt.NDArray[np.float64], npt.NDArray[np.intp], npt.NDArray[np.intp]])
63assert_type(np.unique(AR_LIKE_f8, return_inverse=True, return_counts=True), tuple[npt.NDArray[Any], npt.NDArray[np.intp], npt.NDArray[np.intp]])
64assert_type(np.unique(AR_f8, return_index=True, return_inverse=True, return_counts=True), tuple[npt.NDArray[np.float64], npt.NDArray[np.intp], npt.NDArray[np.intp], npt.NDArray[np.intp]])
65assert_type(np.unique(AR_LIKE_f8, return_index=True, return_inverse=True, return_counts=True), tuple[npt.NDArray[Any], npt.NDArray[np.intp], npt.NDArray[np.intp], npt.NDArray[np.intp]])
66
67assert_type(np.unique_all(AR_f8), UniqueAllResult[np.float64])
68assert_type(np.unique_all(AR_LIKE_f8), UniqueAllResult[Any])
69assert_type(np.unique_counts(AR_f8), UniqueCountsResult[np.float64])
70assert_type(np.unique_counts(AR_LIKE_f8), UniqueCountsResult[Any])
71assert_type(np.unique_inverse(AR_f8), UniqueInverseResult[np.float64])
72assert_type(np.unique_inverse(AR_LIKE_f8), UniqueInverseResult[Any])
73assert_type(np.unique_values(AR_f8), npt.NDArray[np.float64])
74assert_type(np.unique_values(AR_LIKE_f8), npt.NDArray[Any])
75