@@ -530,26 +530,17 @@ def test_keepdims_out(self):
530
530
531
531
532
532
533
-
534
533
class TestAnyGeneric (_GenericReductionsTestMixin , _GenericHasOutTestMixin ):
535
534
def setup_method (self ):
536
535
self .func = np .any
537
536
self .allowed_axes = [0 , 1 , 2 , - 1 , - 2 ,]
538
537
539
- @pytest .mark .xfail (reason = 'XXX: pytorch does not support any(..., axis=tuple)' )
540
- def test_axis_bad_tuple (self ):
541
- super ().test_axis_bad_tuple ()
542
-
543
538
544
539
class TestAllGeneric (_GenericReductionsTestMixin , _GenericHasOutTestMixin ):
545
540
def setup_method (self ):
546
541
self .func = np .all
547
542
self .allowed_axes = [0 , 1 , 2 , - 1 , - 2 ,]
548
543
549
- @pytest .mark .xfail (reason = 'XXX: pytorch does not support all(..., axis=tuple)' )
550
- def test_axis_bad_tuple (self ):
551
- super ().test_axis_bad_tuple ()
552
-
553
544
554
545
class TestCountNonzeroGeneric (_GenericReductionsTestMixin ):
555
546
# count_nonzero does not have the out=... argument
@@ -564,32 +555,12 @@ def setup_method(self):
564
555
self .func = np .argmin
565
556
self .allowed_axes = [0 , 1 , 2 , - 1 , - 2 , ]
566
557
567
- @pytest .mark .xfail (reason = 'XXX: argmin does not allow axis=tuple' )
568
- def test_axis_bad_tuple (self ):
569
- # argmin reject axis tuples in both numpy and pytorch
570
- super ().test_axis_bad_tuple ()
571
-
572
- @pytest .mark .xfail (reason = 'XXX: argmin does not allow axis=tuple)' )
573
- def test_axis_empty_generic (self ):
574
- # argmin reject axis tuples in both numpy and pytorch
575
- super ().test_axis_empty_generic ()
576
-
577
558
578
559
class TestArgmaxGeneric (_GenericReductionsTestMixin , _GenericHasOutTestMixin ):
579
560
def setup_method (self ):
580
561
self .func = np .argmax
581
562
self .allowed_axes = [0 , 1 , 2 , - 1 , - 2 , ]
582
563
583
- @pytest .mark .xfail (reason = 'XXX: argmax does not allow axis=tuple' )
584
- def test_axis_bad_tuple (self ):
585
- # argmax reject axis tuples in both numpy and pytorch
586
- super ().test_axis_bad_tuple ()
587
-
588
- @pytest .mark .xfail (reason = 'XXX: argmax does not allow axis=tuple)' )
589
- def test_axis_empty_generic (self ):
590
- # argmax reject axis tuples in both numpy and pytorch
591
- super ().test_axis_empty_generic ()
592
-
593
564
594
565
class TestAmaxGeneric (_GenericReductionsTestMixin , _GenericHasOutTestMixin ):
595
566
def setup_method (self ):
0 commit comments