@@ -914,18 +914,24 @@ def test_regrids_max_masked(self):
914914 )
915915 )
916916
917- x = s .regrids (d , method = "linear" , use_dst_mask = False , max_masked = 3 )
918- self .assertTrue (
919- np .array_equal (
920- x .data .mask ,
921- [
922- [False , False , True , True , True , True , True , True ],
923- [False , False , False , True , True , True , True , True ],
924- [False , False , False , False , True , True , True , True ],
925- [False , False , False , False , False , True , True , True ],
926- ],
917+ for n in (3 , 4 , 5 ):
918+ x = s .regrids (d , method = "linear" , use_dst_mask = False , max_masked = n )
919+ self .assertTrue (
920+ np .array_equal (
921+ x .data .mask ,
922+ [
923+ [False , False , True , True , True , True , True , True ],
924+ [False , False , False , True , True , True , True , True ],
925+ [False , False , False , False , True , True , True , True ],
926+ [False , False , False , False , False , True , True , True ],
927+ ],
928+ )
927929 )
928- )
930+
931+ # Check bad values of max_masked
932+ for n in (- 1 , 3.14 , "string" , None ):
933+ with self .assertRaises (ValueError ):
934+ x = s .regrids (d , method = "linear" , max_masked = n )
929935
930936 def test_regridc_max_masked (self ):
931937 """Test max_masked keyword to regridc."""
@@ -1011,20 +1017,21 @@ def test_regridc_max_masked(self):
10111017 )
10121018 )
10131019
1014- x = s .regridc (
1015- d , axes = axes , method = "linear" , use_dst_mask = False , max_masked = 3
1016- )
1017- self .assertTrue (
1018- np .array_equal (
1019- x .data .mask ,
1020- [
1021- [True , False , True , True , True , True , True , True ],
1022- [True , False , False , True , True , True , True , True ],
1023- [True , False , False , False , True , True , True , True ],
1024- [True , False , False , False , False , True , True , True ],
1025- ],
1020+ for n in (3 , 4 , 5 ):
1021+ x = s .regridc (
1022+ d , axes = axes , method = "linear" , use_dst_mask = False , max_masked = n
1023+ )
1024+ self .assertTrue (
1025+ np .array_equal (
1026+ x .data .mask ,
1027+ [
1028+ [True , False , True , True , True , True , True , True ],
1029+ [True , False , False , True , True , True , True , True ],
1030+ [True , False , False , False , True , True , True , True ],
1031+ [True , False , False , False , False , True , True , True ],
1032+ ],
1033+ )
10261034 )
1027- )
10281035
10291036
10301037if __name__ == "__main__" :
0 commit comments