Skip to content

Commit 37e0c2e

Browse files
committed
✅ update type-tests
1 parent 21673e6 commit 37e0c2e

File tree

4 files changed

+12
-14
lines changed

4 files changed

+12
-14
lines changed

src/numpy-stubs/@test/runtime/legacy/multiarray.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,7 @@
7070
np.unpackbits(AR_u1)
7171

7272
np.shares_memory(1, 2)
73-
np.shares_memory(AR_f8, AR_f8, max_work=1)
73+
np.shares_memory(AR_f8, AR_f8, max_work=-1)
7474

7575
np.may_share_memory(1, 2)
76-
np.may_share_memory(AR_f8, AR_f8, max_work=1)
76+
np.may_share_memory(AR_f8, AR_f8, max_work=-1)

src/numpy-stubs/@test/static/accept/array_constructors.pyi

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -102,15 +102,15 @@ assert_type(np.frombuffer(A, dtype="c16"), _nt.Array)
102102
assert_type(np.arange(False, True), _nt.Array1D[np.int_])
103103
assert_type(np.arange(10), _nt.Array1D[np.int_])
104104
assert_type(np.arange(0, 10, step=2), _nt.Array1D[np.int_])
105-
assert_type(np.arange(10.0), _nt.Array1D[np.float64])
106-
assert_type(np.arange(start=0, stop=10.0), _nt.Array1D[np.float64])
107-
assert_type(np.arange(np.timedelta64(0)), _nt.Array1D[np.timedelta64])
108-
assert_type(np.arange(0, np.timedelta64(10)), _nt.Array1D[np.timedelta64])
109-
assert_type(np.arange(np.datetime64("0"), np.datetime64("10")), _nt.Array1D[np.datetime64])
110-
assert_type(np.arange(10, dtype=np.float64), _nt.Array1D[np.float64])
105+
assert_type(np.arange(10.0), _nt.Array1D[np.float64 | Any])
106+
assert_type(np.arange(0, stop=10.0), _nt.Array1D[np.float64 | Any])
107+
assert_type(np.arange(np.timedelta64(0)), _nt.Array1D[np.timedelta64[Any]])
108+
assert_type(np.arange(0, np.timedelta64(10)), _nt.Array1D[np.timedelta64[Any]])
109+
assert_type(np.arange(np.datetime64("0"), np.datetime64("10")), _nt.Array1D[np.datetime64[Any]])
110+
assert_type(np.arange(10, dtype=np.float64), _nt.Array1D[np.float64 | Any])
111111
assert_type(np.arange(0, 10, step=2, dtype=np.int16), _nt.Array1D[np.int16])
112112
assert_type(np.arange(10, dtype=int), _nt.Array1D[np.int_])
113-
assert_type(np.arange(0, 10, dtype="f8"), _nt.Array1D[np.float64])
113+
assert_type(np.arange(0, 10, dtype="f8"), _nt.Array1D[Any])
114114

115115
assert_type(np.require(A), _nt.Array[np.float64])
116116
assert_type(np.require(B), MyArray[np.float64])

src/numpy-stubs/@test/static/accept/multiarray.pyi

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -101,10 +101,10 @@ assert_type(np.packbits(AR_u1), _nt.Array[np.uint8])
101101
assert_type(np.unpackbits(AR_u1), _nt.Array[np.uint8])
102102

103103
assert_type(np.shares_memory(1, 2), bool)
104-
assert_type(np.shares_memory(AR_f8, AR_f8, max_work=1), bool)
104+
assert_type(np.shares_memory(AR_f8, AR_f8, max_work=-1), bool)
105105

106106
assert_type(np.may_share_memory(1, 2), bool)
107-
assert_type(np.may_share_memory(AR_f8, AR_f8, max_work=1), bool)
107+
assert_type(np.may_share_memory(AR_f8, AR_f8, max_work=-1), bool)
108108

109109
assert_type(np.promote_types(np.int32, np.int64), np.dtype)
110110
assert_type(np.promote_types("f4", float), np.dtype)

src/numpy-stubs/@test/static/reject/multiarray.pyi

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ def func(a: int) -> None: ...
1616

1717
###
1818

19-
np.where(AR_b, 1) # type: ignore[call-overload] # pyright: ignore[reportCallIssue]
19+
np.where(AR_b, 1) # type: ignore[call-overload] # pyright: ignore[reportArgumentType]
2020

2121
np.can_cast(AR_f8, 1) # type: ignore[arg-type] # pyright: ignore[reportArgumentType]
2222

@@ -35,8 +35,6 @@ np.unpackbits(AR_u1, bitorder=">") # type: ignore[arg-type] # pyright: ignore[
3535
np.shares_memory(1, 1, max_work=i8) # type: ignore[arg-type] # pyright: ignore[reportArgumentType]
3636
np.may_share_memory(1, 1, max_work=i8) # type: ignore[arg-type] # pyright: ignore[reportArgumentType]
3737

38-
np.arange(M) # type: ignore[call-overload] # pyright: ignore[reportArgumentType, reportCallIssue]
39-
4038
np.datetime_data(int) # type: ignore[arg-type] # pyright: ignore[reportArgumentType]
4139
np.datetime_as_string("2012") # type: ignore[call-overload] # pyright: ignore[reportArgumentType, reportCallIssue]
4240

0 commit comments

Comments
 (0)