File tree Expand file tree Collapse file tree 1 file changed +9
-4
lines changed Expand file tree Collapse file tree 1 file changed +9
-4
lines changed Original file line number Diff line number Diff line change 22
22
23
23
class Dummy :
24
24
@staticmethod
25
- def __abs__ (a ):
25
+ def abs (a ):
26
26
return a
27
27
28
28
@staticmethod
29
- def __add__ (a , b ):
29
+ def add (a , b ):
30
30
if isinstance (a , dpt .usm_ndarray ):
31
31
return a
32
32
else :
33
33
return b
34
34
35
35
@staticmethod
36
- def __sub__ (a , b ):
36
+ def subtract (a , b ):
37
37
if isinstance (a , dpt .usm_ndarray ):
38
38
return a
39
39
else :
40
40
return b
41
41
42
42
@staticmethod
43
- def __mul__ (a , b ):
43
+ def multiply (a , b ):
44
44
if isinstance (a , dpt .usm_ndarray ):
45
45
return a
46
46
else :
@@ -51,6 +51,7 @@ def __mul__(a, b):
51
51
def test_fp_ops (namespace ):
52
52
X = dpt .usm_ndarray (1 , "d" )
53
53
X ._set_namespace (namespace )
54
+ assert X .__array_namespace__ () is namespace
54
55
X [0 ] = - 2.5
55
56
X .__abs__ ()
56
57
X .__add__ (1.0 )
@@ -75,8 +76,10 @@ def test_fp_ops(namespace):
75
76
X .__imul__ (X )
76
77
X .__itruediv__ (1.0 )
77
78
X .__ifloordiv__ (1.0 )
79
+
78
80
X = dpt .usm_ndarray (1 , "i4" )
79
81
X ._set_namespace (namespace )
82
+ assert X .__array_namespace__ () is namespace
80
83
X .__lshift__ (2 )
81
84
X .__rshift__ (2 )
82
85
X .__rlshift__ (2 )
@@ -99,8 +102,10 @@ def test_fp_ops(namespace):
99
102
X .__pow__ (2 )
100
103
X .__rpow__ (2 )
101
104
X .__ipow__ (2 )
105
+
102
106
M = dpt .from_numpy (np .eye (3 , 3 , dtype = "d" ))
103
107
X ._set_namespace (namespace )
108
+ assert X .__array_namespace__ () is namespace
104
109
M .__matmul__ (M )
105
110
M .__imatmul__ (M )
106
111
M .__rmatmul__ (M )
You can’t perform that action at this time.
0 commit comments