Commitc51aeef

committed

Fixed bug in find_nearest_contour; Added test

1 parent6c3412b commitc51aeefCopy full SHA for c51aeef

File tree

+23

-1

lines changed

+23

-1

lines changed

Lines changed: 1 addition & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -1371,7 +1371,7 @@ def find_nearest_contour(self, x, y, indices=None, pixel=True):`
`1371`	`1371`	`# Nonetheless, improvements could probably be made.`
`1372`	`1372`
`1373`	`1373`	`ifindicesisNone:`
`1374`		`-indices=range(len(self.levels))`
	`1374`	`+indices=range(len(self.layers))`
`1375`	`1375`
`1376`	`1376`	`d2min=np.inf`
`1377`	`1377`	`conmin=None`

Lines changed: 22 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -470,6 +470,28 @@ def test_contour_line_start_on_corner_edge():`
`470`	`470`	`cbar.add_lines(lines)`
`471`	`471`
`472`	`472`
	`473`	`+deftest_find_nearest_contour():`
	`474`	`+xy=np.indices((10,10))`
	`475`	`+img=np.exp(-np.pi(np.sum((xy-5)2,0)/5.*2))`
	`476`	`+cs=plt.contourf(img,10)`
	`477`	`+`
	`478`	`+nearest_contour=cs.find_nearest_contour(1,1,pixel=False)`
	`479`	`+expected_nearest= (0,0,1,1.0,0.0,1.0)`
	`480`	`+assert_array_almost_equal(nearest_contour,expected_nearest)`
	`481`	`+`
	`482`	`+nearest_contour=cs.find_nearest_contour(8,1,pixel=False)`
	`483`	`+expected_nearest= (0,0,16,7.550173,1.587542,0.547551)`
	`484`	`+assert_array_almost_equal(nearest_contour,expected_nearest)`
	`485`	`+`
	`486`	`+nearest_contour=cs.find_nearest_contour(2,5,pixel=False)`
	`487`	`+expected_nearest= (2,0,35,1.884384,4.976664,0.013911)`
	`488`	`+assert_array_almost_equal(nearest_contour,expected_nearest)`
	`489`	`+`
	`490`	`+nearest_contour=cs.find_nearest_contour(2,5,indices=(5,7),pixel=False)`
	`491`	`+expected_nearest= (5,0,16,2.628202,5.0,0.394638)`
	`492`	`+assert_array_almost_equal(nearest_contour,expected_nearest)`
	`493`	`+`
	`494`	`+`
`473`	`495`	`@mpl.style.context("default")`
`474`	`496`	`deftest_contour_autolabel_beyond_powerlimits():`
`475`	`497`	`ax=plt.figure().add_subplot()`

Comments

(0)