4
4
from dpctl .tensor ._tensor_impl import default_device_complex_type
5
5
6
6
import dpnp
7
+ from dpnp .tests .helper import (
8
+ has_support_aspect64 ,
9
+ is_win_platform ,
10
+ numpy_version ,
11
+ )
7
12
8
13
info = dpnp .__array_namespace_info__ ()
9
14
default_device = select_default_device ()
@@ -33,8 +38,12 @@ def test_default_dtypes():
33
38
== default_device_complex_type (default_device )
34
39
== dpnp .asarray (0.0j ).dtype
35
40
)
36
- assert dtypes ["integral" ] == dpnp .intp == dpnp .asarray (0 ).dtype
37
- assert dtypes ["indexing" ] == dpnp .intp == dpnp .argmax (dpnp .zeros (10 )).dtype
41
+ if not is_win_platform () or numpy_version () >= "2.0.0" :
42
+ # numpy changed default integer on Windows since 2.0
43
+ assert dtypes ["integral" ] == dpnp .intp == dpnp .asarray (0 ).dtype
44
+ assert (
45
+ dtypes ["indexing" ] == dpnp .intp == dpnp .argmax (dpnp .zeros (10 )).dtype
46
+ )
38
47
39
48
with pytest .raises (TypeError , match = "Expected type" ):
40
49
info .default_dtypes (device = "gpu" )
@@ -55,12 +64,12 @@ def test_dtypes_all():
55
64
"uint64" : numpy .uint64 , # TODO: replace with dpnp.uint64
56
65
"float32" : dpnp .float32 ,
57
66
}
58
- | ({"float64" : dpnp .float64 } if default_device . has_aspect_fp64 else {})
67
+ | ({"float64" : dpnp .float64 } if has_support_aspect64 () else {})
59
68
| {"complex64" : dpnp .complex64 }
60
69
|
61
70
# TODO: update once dpctl-1977 is resolved
62
71
{"complex128" : dpnp .complex128 }
63
- # ({"complex128": dpnp.complex128} if default_device.has_aspect_fp64 else {})
72
+ # ({"complex128": dpnp.complex128} if has_support_aspect64() else {})
64
73
)
65
74
66
75
@@ -80,11 +89,11 @@ def test_dtypes_all():
80
89
},
81
90
"integral" : ("signed integer" , "unsigned integer" ),
82
91
"real floating" : {"float32" : dpnp .float32 }
83
- | ({"float64" : dpnp .float64 } if default_device . has_aspect_fp64 else {}),
92
+ | ({"float64" : dpnp .float64 } if has_support_aspect64 () else {}),
84
93
"complex floating" : {"complex64" : dpnp .complex64 } |
85
94
# TODO: update once dpctl-1977 is resolved
86
95
{"complex128" : dpnp .complex128 },
87
- # ({"complex128": dpnp.complex128} if default_device.has_aspect_fp64 else {}),
96
+ # ({"complex128": dpnp.complex128} if has_support_aspect64() else {}),
88
97
"numeric" : ("integral" , "real floating" , "complex floating" ),
89
98
}
90
99
0 commit comments