Commit6934853

committed

Added test for cmap.n_variates == 0 in colorizer

1 parent9409d4c commit6934853Copy full SHA for 6934853

File tree

+14

-1

lines changed

+14

-1

lines changed

Lines changed: 4 additions & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -766,7 +766,7 @@ def _ensure_norm(norm, n_components=1):`
`766`	`766`	`scale_cls=_api.check_getitem(scale._scale_mapping,norm=norm)`
`767`	`767`	`return_auto_norm_from_scale(scale_cls)()`
`768`	`768`	`returnnorm`
`769`		`-else:#n_components > 1`
	`769`	`+elifn_components>1:`
`770`	`770`	`ifnotnp.iterable(norm):`
`771`	`771`	`_api.check_isinstance((colors.MultiNorm,None,tuple),norm=norm)`
`772`	`772`	`ifnormisNone:`
`@@ -777,6 +777,9 @@ def _ensure_norm(norm, n_components=1):`
`777`	`777`	`returnnorm`
`778`	`778`	`raiseValueError(`
`779`	`779`	`f"Invalid norm for multivariate colormap with{n_components} inputs")`
	`780`	`+else:# n_components == 0`
	`781`	`+raiseValueError(`
	`782`	+"Invalid cmap. A colorizer object must have a cmap with `n_variates` >= 1")
`780`	`783`
`781`	`784`
`782`	`785`	`def_ensure_cmap(cmap,accept_multivariate=False):`

Lines changed: 10 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -2199,6 +2199,16 @@ def test_colorizer_multinorm_explicit():`
`2199`	`2199`	`assert_array_almost_equal(ca.to_rgba(data),res)`
`2200`	`2200`
`2201`	`2201`
	`2202`	`+deftest_invalid_cmap_n_components_zero():`
	`2203`	`+classCustomColormap(mcolors.Colormap):`
	`2204`	`+def__init__(self):`
	`2205`	`+super().__init__("custom")`
	`2206`	`+self.n_variates=0`
	`2207`	`+`
	`2208`	+withpytest.raises(ValueError,match='`n_variates` >= 1'):
	`2209`	`+ca=mcolorizer.Colorizer(CustomColormap())`
	`2210`	`+`
	`2211`	`+`
`2202`	`2212`	`deftest_colorizer_bivar_cmap():`
`2203`	`2213`	`ca=mcolorizer.Colorizer('BiOrangeBlue', [mcolors.Normalize(),'log'])`
`2204`	`2214`

Comments

(0)