Commitf94d64d

committed

MAINT: Adding an equals method to colormaps

This adds an __eq__ method to Colormap to test the lookuptables for equality.

1 parentcb21d7d commitf94d64dCopy full SHA for f94d64d

File tree

+25

-0

lines changed

+25

-0

lines changed

Lines changed: 10 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -690,6 +690,16 @@ def __copy__(self):`
`690`	`690`	`cmapobject._global=False`
`691`	`691`	`returncmapobject`
`692`	`692`
	`693`	`+def__eq__(self,other):`
	`694`	`+ifnotisinstance(other,Colormap):`
	`695`	`+returnFalse`
	`696`	`+# To compare lookup tables the Colormaps have to be initialized`
	`697`	`+ifnotself._isinit:`
	`698`	`+self._init()`
	`699`	`+ifnotother._isinit:`
	`700`	`+other._init()`
	`701`	`+returnnp.array_equal(self._lut,other._lut)`
	`702`	`+`
`693`	`703`	`defget_bad(self):`
`694`	`704`	`"""Get the color for masked values."""`
`695`	`705`	`ifnotself._isinit:`

Lines changed: 15 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -162,6 +162,21 @@ def test_colormap_copy():`
`162`	`162`	`assert_array_equal(ret1,ret2)`
`163`	`163`
`164`	`164`
	`165`	`+deftest_colormap_equals():`
	`166`	`+cmap=plt.get_cmap("plasma")`
	`167`	`+cm_copy=cmap.copy()`
	`168`	`+# different object id's`
	`169`	`+assertcm_copyisnotcmap`
	`170`	`+# But the same data should be equal`
	`171`	`+assertcm_copy==cmap`
	`172`	`+# Change the copy`
	`173`	`+cm_copy.set_bad('y')`
	`174`	`+assertcm_copy!=cmap`
	`175`	`+# Make sure we can compare different sizes without failure`
	`176`	`+cm_copy._lut=cm_copy._lut[:10, :]`
	`177`	`+assertcm_copy!=cmap`
	`178`	`+`
	`179`	`+`
`165`	`180`	`deftest_colormap_endian():`
`166`	`181`	`"""`
`167`	`182`	`GitHub issue #1005: a bug in putmask caused erroneous`

Comments

(0)