@@ -58,9 +58,9 @@ def test_hist():
58
58
assert isinstance (h [0 , :], cuda_histogram .Hist )
59
59
assert isinstance (h [0 , :].values (), cp .ndarray )
60
60
assert (h [0 , :].values () == h .values ()[0 , :]).all ()
61
- assert isinstance (h [0 ], cuda_histogram .Hist )
62
- assert isinstance (h [0 ].values (), cp .ndarray )
63
- assert (h [0 ].values () == h .values ()[0 ]).all ()
61
+ assert isinstance (h [0 , : ], cuda_histogram .Hist )
62
+ assert isinstance (h [0 , : ].values (), cp .ndarray )
63
+ assert (h [0 , : ].values () == h .values ()[0 ]).all ()
64
64
assert isinstance (h [:, 0 ], cuda_histogram .Hist )
65
65
assert isinstance (h [:, 0 ].values (), cp .ndarray )
66
66
assert (h [:, 0 ].values () == h .values ()[:, 0 ].reshape ((20 , 1 ))).all ()
@@ -99,11 +99,11 @@ def test_hist():
99
99
assert isinstance (h [0 , :].variance (), cp .ndarray )
100
100
assert (h [0 , :].values () == h .values ()[0 , :]).all ()
101
101
assert (h [0 , :].variance () == h .variance ()[0 , :]).all ()
102
- assert isinstance (h [0 ], cuda_histogram .Hist )
103
- assert isinstance (h [0 ].values (), cp .ndarray )
104
- assert isinstance (h [0 ].variance (), cp .ndarray )
105
- assert (h [0 ].values () == h .values ()[0 ]).all ()
106
- assert (h [0 ].variance () == h .variance ()[0 ]).all ()
102
+ assert isinstance (h [0 , : ], cuda_histogram .Hist )
103
+ assert isinstance (h [0 , : ].values (), cp .ndarray )
104
+ assert isinstance (h [0 , : ].variance (), cp .ndarray )
105
+ assert (h [0 , : ].values () == h .values ()[0 ]).all ()
106
+ assert (h [0 , : ].variance () == h .variance ()[0 ]).all ()
107
107
assert isinstance (h [:, 0 ], cuda_histogram .Hist )
108
108
assert isinstance (h [:, 0 ].values (), cp .ndarray )
109
109
assert isinstance (h [:, 0 ].variance (), cp .ndarray )
0 commit comments