def test_abundance_matplotlib_default_colors():
# Matplotlib colors:
ax = pp.abundance(Q,
pressure,
species,
colors=None,
highlight='H2O CH4 CO CO2 NH3 HCN H2 H He'.split())
def test_abundance_custom_colors():
colors = 'b r g y m k 0.5'.split()
ax = pp.abundance(Q,
pressure,
species,
colors=colors,
highlight='H2O CH4 CO CO2 NH3 HCN H2 H He'.split())
def test_abundance_one_to_one_dashes_and_colors_alternative_highlight():
colors = 'b r g y m k 0.5'.split()
ax = pp.abundance(Q,
pressure,
species[:7],
colors=colors,
dashes=[(), (3, 1), (), (), (), (), ()],
highlight='CH4 CO CO2 NH3 HCN H2 H He'.split())
def test_abundance_one_to_one_colors_alternative_highlight():
# ncolors >= nspecies, different highlight preserves coloring:
colors = 'b r g y m k 0.5'.split()
ax = pp.abundance(Q,
pressure,
species[:7],
colors=colors,
highlight='CH4 CO CO2 NH3 HCN H2 H He'.split())
def test_abundance_one_to_one_colors():
# Custom colors, ncolors == nspecies:
colors = 'b r g y m k 0.5'.split()
ax = pp.abundance(Q,
pressure,
species[:7],
colors=colors,
highlight='H2O CH4 CO CO2 NH3 HCN H2 H He'.split())
def test_abundance_custom_colors_a():
# Different highlight changes color assignments:
colors = 'b r g y m k 0.5'.split()
ax = pp.abundance(Q,
pressure,
species,
colors=colors,
highlight='CH4 CO CO2 NH3 HCN H2 H He'.split())
def test_abundance_custom_dashes_and_colors():
colors = 'b r g y m k 0.5'.split()
ax = pp.abundance(Q,
pressure,
species,
colors=colors,
dashes=cycler(dashes=[(8, 1), (3, 1)]),
highlight='H2O CH4 CO CO2 NH3 HCN H2 H He'.split())
def test_abundance_defaults_extra_molecs():
spec2 = np.copy(species)
spec2[1] = 'CH3'
spec2[2] = 'O'
ax = pp.abundance(Q,
pressure,
spec2,
colors='default',
highlight='H2O CH4 CO CO2 NH3 HCN H2 H He'.split())
def test_abundance_default_molecs():
ax = pp.abundance(Q,
pressure,
species,
colors='default',
highlight='H2O CH4 CO CO2 NH3 HCN H2 H He'.split())