@@ -58,9 +58,9 @@ def test_hist():
5858 assert isinstance (h [0 , :], cuda_histogram .Hist )
5959 assert isinstance (h [0 , :].values (), cp .ndarray )
6060 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 ()
6464 assert isinstance (h [:, 0 ], cuda_histogram .Hist )
6565 assert isinstance (h [:, 0 ].values (), cp .ndarray )
6666 assert (h [:, 0 ].values () == h .values ()[:, 0 ].reshape ((20 , 1 ))).all ()
@@ -99,11 +99,11 @@ def test_hist():
9999 assert isinstance (h [0 , :].variance (), cp .ndarray )
100100 assert (h [0 , :].values () == h .values ()[0 , :]).all ()
101101 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 ()
107107 assert isinstance (h [:, 0 ], cuda_histogram .Hist )
108108 assert isinstance (h [:, 0 ].values (), cp .ndarray )
109109 assert isinstance (h [:, 0 ].variance (), cp .ndarray )
0 commit comments