@@ -195,14 +195,24 @@ def write_dense(uri):
195195 bounded_anytuple (min_value = 0 ,max_value = 99 ),
196196 bounded_anytuple (min_value = 1 ,max_value = 100 )
197197 )
198- def test_ref_2d_slicing (self , dense_array_2d , r1 , r2 ):
198+ def test_ref_2d_points (self , dense_array_2d , pts1 , pts2 ):
199199 a1_legacy , a1_ref = dense_array_2d
200200
201- assume ((r1 ,r2 ) != ((0 ,),(71 , 1 , 72 )))
202- assume ((r1 ,r2 ) != ((0 ,),(1 ,3 ,2 )))
203-
204201 #print(r1,r2)
205202 np .testing .assert_array_equal (
206- a1_legacy .multi_index [list (r1 ),list (r2 )]["a1" ],
207- a1_ref .multi_index [list (r1 ),list (r2 )]["a1" ]
203+ a1_legacy .multi_index [list (pts1 ),list (pts2 )]["a1" ],
204+ a1_ref .multi_index [list (pts1 ),list (pts2 )]["a1" ]
205+ )
206+
207+ @given (
208+ st .lists (ranged_slices (min_value = 0 , max_value = 99 )).filter (lambda x : x != []),
209+ st .lists (ranged_slices (min_value = 1 ,max_value = 100 ).filter (lambda x : x != []))
210+ )
211+ def test_ref_2d_slicing (self , dense_array_2d , ranges1 , ranges2 ):
212+ a1_legacy , a1_ref = dense_array_2d
213+
214+ #print(ranges1, ranges2)
215+ np .testing .assert_array_equal (
216+ a1_legacy .multi_index [list (ranges1 ),list (ranges2 )]["a1" ],
217+ a1_ref .multi_index [list (ranges1 ),list (ranges2 )]["a1" ]
208218 )
0 commit comments