Commite05a632

committed

Improve test for correct clipping

1 parent068fa28 commite05a632Copy full SHA for e05a632

File tree

-3

lines changed

-3

lines changed

Lines changed: 5 additions & 3 deletions

Original file line number	Diff line number	Diff line change
`@@ -801,16 +801,18 @@ def test_imshow_clips_rgb_to_valid_range(dtype):`
`801`	`801`	`arr=np.arange(300,dtype=dtype).reshape((10,10,3))`
`802`	`802`	`ifdtype.kind!='u':`
`803`	`803`	`arr-=10`
	`804`	`+too_low=arr<0`
	`805`	`+too_high=arr>255`
`804`	`806`	`ifdtype.kind=='f':`
`805`	`807`	`arr=arr/255`
`806`	`808`	`_,ax=plt.subplots()`
`807`	`809`	`out=ax.imshow(arr).get_array()`
`808`		`-assertout.min()==0`
	`810`	`+assert(out[too_low]==0).all()`
`809`	`811`	`ifdtype.kind=='f':`
`810`		`-assertout.max()==1`
	`812`	`+assert(out[too_high]==1).all()`
`811`	`813`	`assertout.dtype.kind=='f'`
`812`	`814`	`else:`
`813`		`-assertout.max()==255`
	`815`	`+assert(out[too_high]==255).all()`
`814`	`816`	`assertout.dtype==np.uint8`
`815`	`817`
`816`	`818`

Comments

(0)