@@ -3633,11 +3633,28 @@ def mean_only_one_axis(x, axis):
36333633 actual = ds .reduce (mean_only_one_axis , 'y' )
36343634 assert_identical (expected , actual )
36353635
3636- with raises_regex (TypeError , 'non-integer axis' ):
3636+ with raises_regex (TypeError , "missing 1 required positional argument: "
3637+ "'axis'" ):
36373638 ds .reduce (mean_only_one_axis )
36383639
36393640 with raises_regex (TypeError , 'non-integer axis' ):
3640- ds .reduce (mean_only_one_axis , ['x' , 'y' ])
3641+ ds .reduce (mean_only_one_axis , axis = ['x' , 'y' ])
3642+
3643+ def test_reduce_no_axis (self ):
3644+
3645+ def total_sum (x ):
3646+ return np .sum (x .flatten ())
3647+
3648+ ds = Dataset ({'a' : (['x' , 'y' ], [[0 , 1 , 2 , 3 , 4 ]])})
3649+ expected = Dataset ({'a' : ((), 10 )})
3650+ actual = ds .reduce (total_sum )
3651+ assert_identical (expected , actual )
3652+
3653+ with raises_regex (TypeError , "unexpected keyword argument 'axis'" ):
3654+ ds .reduce (total_sum , axis = 0 )
3655+
3656+ with raises_regex (TypeError , "unexpected keyword argument 'axis'" ):
3657+ ds .reduce (total_sum , dim = 'x' )
36413658
36423659 def test_quantile (self ):
36433660
0 commit comments