12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307 |
- try:
- from contextlib import nullcontext
- except ImportError:
- from contextlib import ExitStack as nullcontext # Py 3.6.
- import re
- import itertools
- import numpy as np
- from numpy.testing import assert_almost_equal, assert_array_equal
- import pytest
- import matplotlib
- import matplotlib.pyplot as plt
- import matplotlib.ticker as mticker
- class TestMaxNLocator:
- basic_data = [
- (20, 100, np.array([20., 40., 60., 80., 100.])),
- (0.001, 0.0001, np.array([0., 0.0002, 0.0004, 0.0006, 0.0008, 0.001])),
- (-1e15, 1e15, np.array([-1.0e+15, -5.0e+14, 0e+00, 5e+14, 1.0e+15])),
- (0, 0.85e-50, np.arange(6) * 2e-51),
- (-0.85e-50, 0, np.arange(-5, 1) * 2e-51),
- ]
- integer_data = [
- (-0.1, 1.1, None, np.array([-1, 0, 1, 2])),
- (-0.1, 0.95, None, np.array([-0.25, 0, 0.25, 0.5, 0.75, 1.0])),
- (1, 55, [1, 1.5, 5, 6, 10], np.array([0, 15, 30, 45, 60])),
- ]
- @pytest.mark.parametrize('vmin, vmax, expected', basic_data)
- def test_basic(self, vmin, vmax, expected):
- loc = mticker.MaxNLocator(nbins=5)
- assert_almost_equal(loc.tick_values(vmin, vmax), expected)
- @pytest.mark.parametrize('vmin, vmax, steps, expected', integer_data)
- def test_integer(self, vmin, vmax, steps, expected):
- loc = mticker.MaxNLocator(nbins=5, integer=True, steps=steps)
- assert_almost_equal(loc.tick_values(vmin, vmax), expected)
- class TestLinearLocator:
- def test_basic(self):
- loc = mticker.LinearLocator(numticks=3)
- test_value = np.array([-0.8, -0.3, 0.2])
- assert_almost_equal(loc.tick_values(-0.8, 0.2), test_value)
- def test_set_params(self):
- """
- Create linear locator with presets={}, numticks=2 and change it to
- something else. See if change was successful. Should not exception.
- """
- loc = mticker.LinearLocator(numticks=2)
- loc.set_params(numticks=8, presets={(0, 1): []})
- assert loc.numticks == 8
- assert loc.presets == {(0, 1): []}
- class TestMultipleLocator:
- def test_basic(self):
- loc = mticker.MultipleLocator(base=3.147)
- test_value = np.array([-9.441, -6.294, -3.147, 0., 3.147, 6.294,
- 9.441, 12.588])
- assert_almost_equal(loc.tick_values(-7, 10), test_value)
- def test_view_limits(self):
- """
- Test basic behavior of view limits.
- """
- with matplotlib.rc_context({'axes.autolimit_mode': 'data'}):
- loc = mticker.MultipleLocator(base=3.147)
- assert_almost_equal(loc.view_limits(-5, 5), (-5, 5))
- def test_view_limits_round_numbers(self):
- """
- Test that everything works properly with 'round_numbers' for auto
- limit.
- """
- with matplotlib.rc_context({'axes.autolimit_mode': 'round_numbers'}):
- loc = mticker.MultipleLocator(base=3.147)
- assert_almost_equal(loc.view_limits(-4, 4), (-6.294, 6.294))
- def test_set_params(self):
- """
- Create multiple locator with 0.7 base, and change it to something else.
- See if change was successful.
- """
- mult = mticker.MultipleLocator(base=0.7)
- mult.set_params(base=1.7)
- assert mult._edge.step == 1.7
- class TestAutoMinorLocator:
- def test_basic(self):
- fig, ax = plt.subplots()
- ax.set_xlim(0, 1.39)
- ax.minorticks_on()
- test_value = np.array([0.05, 0.1, 0.15, 0.25, 0.3, 0.35, 0.45,
- 0.5, 0.55, 0.65, 0.7, 0.75, 0.85, 0.9,
- 0.95, 1.05, 1.1, 1.15, 1.25, 1.3, 1.35])
- assert_almost_equal(ax.xaxis.get_ticklocs(minor=True), test_value)
- # NB: the following values are assuming that *xlim* is [0, 5]
- params = [
- (0, 0), # no major tick => no minor tick either
- (1, 0) # a single major tick => no minor tick
- ]
- @pytest.mark.parametrize('nb_majorticks, expected_nb_minorticks', params)
- def test_low_number_of_majorticks(
- self, nb_majorticks, expected_nb_minorticks):
- # This test is related to issue #8804
- fig, ax = plt.subplots()
- xlims = (0, 5) # easier to test the different code paths
- ax.set_xlim(*xlims)
- ax.set_xticks(np.linspace(xlims[0], xlims[1], nb_majorticks))
- ax.minorticks_on()
- ax.xaxis.set_minor_locator(mticker.AutoMinorLocator())
- assert len(ax.xaxis.get_minorticklocs()) == expected_nb_minorticks
- majorstep_minordivisions = [(1, 5),
- (2, 4),
- (2.5, 5),
- (5, 5),
- (10, 5)]
- # This test is meant to verify the parameterization for
- # test_number_of_minor_ticks
- def test_using_all_default_major_steps(self):
- with matplotlib.rc_context({'_internal.classic_mode': False}):
- majorsteps = [x[0] for x in self.majorstep_minordivisions]
- np.testing.assert_allclose(majorsteps,
- mticker.AutoLocator()._steps)
- @pytest.mark.parametrize('major_step, expected_nb_minordivisions',
- majorstep_minordivisions)
- def test_number_of_minor_ticks(
- self, major_step, expected_nb_minordivisions):
- fig, ax = plt.subplots()
- xlims = (0, major_step)
- ax.set_xlim(*xlims)
- ax.set_xticks(xlims)
- ax.minorticks_on()
- ax.xaxis.set_minor_locator(mticker.AutoMinorLocator())
- nb_minor_divisions = len(ax.xaxis.get_minorticklocs()) + 1
- assert nb_minor_divisions == expected_nb_minordivisions
- limits = [(0, 1.39), (0, 0.139),
- (0, 0.11e-19), (0, 0.112e-12),
- (-2.0e-07, -3.3e-08), (1.20e-06, 1.42e-06),
- (-1.34e-06, -1.44e-06), (-8.76e-07, -1.51e-06)]
- reference = [
- [0.05, 0.1, 0.15, 0.25, 0.3, 0.35, 0.45, 0.5, 0.55, 0.65, 0.7,
- 0.75, 0.85, 0.9, 0.95, 1.05, 1.1, 1.15, 1.25, 1.3, 1.35],
- [0.005, 0.01, 0.015, 0.025, 0.03, 0.035, 0.045, 0.05, 0.055, 0.065,
- 0.07, 0.075, 0.085, 0.09, 0.095, 0.105, 0.11, 0.115, 0.125, 0.13,
- 0.135],
- [5.00e-22, 1.00e-21, 1.50e-21, 2.50e-21, 3.00e-21, 3.50e-21, 4.50e-21,
- 5.00e-21, 5.50e-21, 6.50e-21, 7.00e-21, 7.50e-21, 8.50e-21, 9.00e-21,
- 9.50e-21, 1.05e-20, 1.10e-20],
- [5.00e-15, 1.00e-14, 1.50e-14, 2.50e-14, 3.00e-14, 3.50e-14, 4.50e-14,
- 5.00e-14, 5.50e-14, 6.50e-14, 7.00e-14, 7.50e-14, 8.50e-14, 9.00e-14,
- 9.50e-14, 1.05e-13, 1.10e-13],
- [-1.95e-07, -1.90e-07, -1.85e-07, -1.75e-07, -1.70e-07, -1.65e-07,
- -1.55e-07, -1.50e-07, -1.45e-07, -1.35e-07, -1.30e-07, -1.25e-07,
- -1.15e-07, -1.10e-07, -1.05e-07, -9.50e-08, -9.00e-08, -8.50e-08,
- -7.50e-08, -7.00e-08, -6.50e-08, -5.50e-08, -5.00e-08, -4.50e-08,
- -3.50e-08],
- [1.21e-06, 1.22e-06, 1.23e-06, 1.24e-06, 1.26e-06, 1.27e-06, 1.28e-06,
- 1.29e-06, 1.31e-06, 1.32e-06, 1.33e-06, 1.34e-06, 1.36e-06, 1.37e-06,
- 1.38e-06, 1.39e-06, 1.41e-06, 1.42e-06],
- [-1.435e-06, -1.430e-06, -1.425e-06, -1.415e-06, -1.410e-06,
- -1.405e-06, -1.395e-06, -1.390e-06, -1.385e-06, -1.375e-06,
- -1.370e-06, -1.365e-06, -1.355e-06, -1.350e-06, -1.345e-06],
- [-1.48e-06, -1.46e-06, -1.44e-06, -1.42e-06, -1.38e-06, -1.36e-06,
- -1.34e-06, -1.32e-06, -1.28e-06, -1.26e-06, -1.24e-06, -1.22e-06,
- -1.18e-06, -1.16e-06, -1.14e-06, -1.12e-06, -1.08e-06, -1.06e-06,
- -1.04e-06, -1.02e-06, -9.80e-07, -9.60e-07, -9.40e-07, -9.20e-07,
- -8.80e-07]]
- additional_data = list(zip(limits, reference))
- @pytest.mark.parametrize('lim, ref', additional_data)
- def test_additional(self, lim, ref):
- fig, ax = plt.subplots()
- ax.minorticks_on()
- ax.grid(True, 'minor', 'y', linewidth=1)
- ax.grid(True, 'major', color='k', linewidth=1)
- ax.set_ylim(lim)
- assert_almost_equal(ax.yaxis.get_ticklocs(minor=True), ref)
- class TestLogLocator:
- def test_basic(self):
- loc = mticker.LogLocator(numticks=5)
- with pytest.raises(ValueError):
- loc.tick_values(0, 1000)
- test_value = np.array([1.00000000e-05, 1.00000000e-03, 1.00000000e-01,
- 1.00000000e+01, 1.00000000e+03, 1.00000000e+05,
- 1.00000000e+07, 1.000000000e+09])
- assert_almost_equal(loc.tick_values(0.001, 1.1e5), test_value)
- loc = mticker.LogLocator(base=2)
- test_value = np.array([0.5, 1., 2., 4., 8., 16., 32., 64., 128., 256.])
- assert_almost_equal(loc.tick_values(1, 100), test_value)
- def test_switch_to_autolocator(self):
- loc = mticker.LogLocator(subs="all")
- assert_array_equal(loc.tick_values(0.45, 0.55),
- [0.44, 0.46, 0.48, 0.5, 0.52, 0.54, 0.56])
- # check that we *skip* 1.0, and 10, because this is a minor locator
- loc = mticker.LogLocator(subs=np.arange(2, 10))
- assert 1.0 not in loc.tick_values(0.9, 20.)
- assert 10.0 not in loc.tick_values(0.9, 20.)
- def test_set_params(self):
- """
- Create log locator with default value, base=10.0, subs=[1.0],
- numdecs=4, numticks=15 and change it to something else.
- See if change was successful. Should not raise exception.
- """
- loc = mticker.LogLocator()
- loc.set_params(numticks=7, numdecs=8, subs=[2.0], base=4)
- assert loc.numticks == 7
- assert loc.numdecs == 8
- assert loc._base == 4
- assert list(loc._subs) == [2.0]
- class TestNullLocator:
- def test_set_params(self):
- """
- Create null locator, and attempt to call set_params() on it.
- Should not exception, and should raise a warning.
- """
- loc = mticker.NullLocator()
- with pytest.warns(UserWarning):
- loc.set_params()
- class _LogitHelper:
- @staticmethod
- def isclose(x, y):
- return (np.isclose(-np.log(1/x-1), -np.log(1/y-1))
- if 0 < x < 1 and 0 < y < 1 else False)
- @staticmethod
- def assert_almost_equal(x, y):
- ax = np.array(x)
- ay = np.array(y)
- assert np.all(ax > 0) and np.all(ax < 1)
- assert np.all(ay > 0) and np.all(ay < 1)
- lx = -np.log(1/ax-1)
- ly = -np.log(1/ay-1)
- assert_almost_equal(lx, ly)
- class TestLogitLocator:
- ref_basic_limits = [
- (5e-2, 1 - 5e-2),
- (5e-3, 1 - 5e-3),
- (5e-4, 1 - 5e-4),
- (5e-5, 1 - 5e-5),
- (5e-6, 1 - 5e-6),
- (5e-7, 1 - 5e-7),
- (5e-8, 1 - 5e-8),
- (5e-9, 1 - 5e-9),
- ]
- ref_basic_major_ticks = [
- 1 / (10 ** np.arange(1, 3)),
- 1 / (10 ** np.arange(1, 4)),
- 1 / (10 ** np.arange(1, 5)),
- 1 / (10 ** np.arange(1, 6)),
- 1 / (10 ** np.arange(1, 7)),
- 1 / (10 ** np.arange(1, 8)),
- 1 / (10 ** np.arange(1, 9)),
- 1 / (10 ** np.arange(1, 10)),
- ]
- ref_maxn_limits = [(0.4, 0.6), (5e-2, 2e-1), (1 - 2e-1, 1 - 5e-2)]
- @pytest.mark.parametrize(
- "lims, expected_low_ticks",
- zip(ref_basic_limits, ref_basic_major_ticks),
- )
- def test_basic_major(self, lims, expected_low_ticks):
- """
- Create logit locator with huge number of major, and tests ticks.
- """
- expected_ticks = sorted(
- [*expected_low_ticks, 0.5, *(1 - expected_low_ticks)]
- )
- loc = mticker.LogitLocator(nbins=100)
- _LogitHelper.assert_almost_equal(
- loc.tick_values(*lims),
- expected_ticks
- )
- @pytest.mark.parametrize("lims", ref_maxn_limits)
- def test_maxn_major(self, lims):
- """
- When the axis is zoomed, the locator must have the same behavior as
- MaxNLocator.
- """
- loc = mticker.LogitLocator(nbins=100)
- maxn_loc = mticker.MaxNLocator(nbins=100, steps=[1, 2, 5, 10])
- for nbins in (4, 8, 16):
- loc.set_params(nbins=nbins)
- maxn_loc.set_params(nbins=nbins)
- ticks = loc.tick_values(*lims)
- maxn_ticks = maxn_loc.tick_values(*lims)
- assert ticks.shape == maxn_ticks.shape
- assert (ticks == maxn_ticks).all()
- @pytest.mark.parametrize("lims", ref_basic_limits + ref_maxn_limits)
- def test_nbins_major(self, lims):
- """
- Assert logit locator for respecting nbins param.
- """
- basic_needed = int(-np.floor(np.log10(lims[0]))) * 2 + 1
- loc = mticker.LogitLocator(nbins=100)
- for nbins in range(basic_needed, 2, -1):
- loc.set_params(nbins=nbins)
- assert len(loc.tick_values(*lims)) <= nbins + 2
- @pytest.mark.parametrize(
- "lims, expected_low_ticks",
- zip(ref_basic_limits, ref_basic_major_ticks),
- )
- def test_minor(self, lims, expected_low_ticks):
- """
- In large scale, test the presence of minor,
- and assert no minor when major are subsampled.
- """
- expected_ticks = sorted(
- [*expected_low_ticks, 0.5, *(1 - expected_low_ticks)]
- )
- basic_needed = len(expected_ticks)
- loc = mticker.LogitLocator(nbins=100)
- minor_loc = mticker.LogitLocator(nbins=100, minor=True)
- for nbins in range(basic_needed, 2, -1):
- loc.set_params(nbins=nbins)
- minor_loc.set_params(nbins=nbins)
- major_ticks = loc.tick_values(*lims)
- minor_ticks = minor_loc.tick_values(*lims)
- if len(major_ticks) >= len(expected_ticks):
- # no subsample, we must have a lot of minors ticks
- assert (len(major_ticks) - 1) * 5 < len(minor_ticks)
- else:
- # subsample
- _LogitHelper.assert_almost_equal(
- np.sort(np.concatenate((major_ticks, minor_ticks))),
- expected_ticks,
- )
- def test_minor_attr(self):
- loc = mticker.LogitLocator(nbins=100)
- assert not loc.minor
- loc.minor = True
- assert loc.minor
- loc.set_params(minor=False)
- assert not loc.minor
- acceptable_vmin_vmax = [
- *(2.5 ** np.arange(-3, 0)),
- *(1 - 2.5 ** np.arange(-3, 0)),
- ]
- @pytest.mark.parametrize(
- "lims",
- [
- (a, b)
- for (a, b) in itertools.product(acceptable_vmin_vmax, repeat=2)
- if a != b
- ],
- )
- def test_nonsingular_ok(self, lims):
- """
- Create logit locator, and test the nonsingular method for acceptable
- value
- """
- loc = mticker.LogitLocator()
- lims2 = loc.nonsingular(*lims)
- assert sorted(lims) == sorted(lims2)
- @pytest.mark.parametrize("okval", acceptable_vmin_vmax)
- def test_nonsingular_nok(self, okval):
- """
- Create logit locator, and test the nonsingular method for non
- acceptable value
- """
- loc = mticker.LogitLocator()
- vmin, vmax = (-1, okval)
- vmin2, vmax2 = loc.nonsingular(vmin, vmax)
- assert vmax2 == vmax
- assert 0 < vmin2 < vmax2
- vmin, vmax = (okval, 2)
- vmin2, vmax2 = loc.nonsingular(vmin, vmax)
- assert vmin2 == vmin
- assert vmin2 < vmax2 < 1
- class TestFixedLocator:
- def test_set_params(self):
- """
- Create fixed locator with 5 nbins, and change it to something else.
- See if change was successful.
- Should not exception.
- """
- fixed = mticker.FixedLocator(range(0, 24), nbins=5)
- fixed.set_params(nbins=7)
- assert fixed.nbins == 7
- class TestIndexLocator:
- def test_set_params(self):
- """
- Create index locator with 3 base, 4 offset. and change it to something
- else. See if change was successful.
- Should not exception.
- """
- index = mticker.IndexLocator(base=3, offset=4)
- index.set_params(base=7, offset=7)
- assert index._base == 7
- assert index.offset == 7
- class TestSymmetricalLogLocator:
- def test_set_params(self):
- """
- Create symmetrical log locator with default subs =[1.0] numticks = 15,
- and change it to something else.
- See if change was successful.
- Should not exception.
- """
- sym = mticker.SymmetricalLogLocator(base=10, linthresh=1)
- sym.set_params(subs=[2.0], numticks=8)
- assert sym._subs == [2.0]
- assert sym.numticks == 8
- class TestScalarFormatter:
- offset_data = [
- (123, 189, 0),
- (-189, -123, 0),
- (12341, 12349, 12340),
- (-12349, -12341, -12340),
- (99999.5, 100010.5, 100000),
- (-100010.5, -99999.5, -100000),
- (99990.5, 100000.5, 100000),
- (-100000.5, -99990.5, -100000),
- (1233999, 1234001, 1234000),
- (-1234001, -1233999, -1234000),
- (1, 1, 1),
- (123, 123, 0),
- # Test cases courtesy of @WeatherGod
- (.4538, .4578, .45),
- (3789.12, 3783.1, 3780),
- (45124.3, 45831.75, 45000),
- (0.000721, 0.0007243, 0.00072),
- (12592.82, 12591.43, 12590),
- (9., 12., 0),
- (900., 1200., 0),
- (1900., 1200., 0),
- (0.99, 1.01, 1),
- (9.99, 10.01, 10),
- (99.99, 100.01, 100),
- (5.99, 6.01, 6),
- (15.99, 16.01, 16),
- (-0.452, 0.492, 0),
- (-0.492, 0.492, 0),
- (12331.4, 12350.5, 12300),
- (-12335.3, 12335.3, 0),
- ]
- use_offset_data = [True, False]
- # (sci_type, scilimits, lim, orderOfMag, fewticks)
- scilimits_data = [
- (False, (0, 0), (10.0, 20.0), 0, False),
- (True, (-2, 2), (-10, 20), 0, False),
- (True, (-2, 2), (-20, 10), 0, False),
- (True, (-2, 2), (-110, 120), 2, False),
- (True, (-2, 2), (-120, 110), 2, False),
- (True, (-2, 2), (-.001, 0.002), -3, False),
- (True, (-7, 7), (0.18e10, 0.83e10), 9, True),
- (True, (0, 0), (-1e5, 1e5), 5, False),
- (True, (6, 6), (-1e5, 1e5), 6, False),
- ]
- @pytest.mark.parametrize('left, right, offset', offset_data)
- def test_offset_value(self, left, right, offset):
- fig, ax = plt.subplots()
- formatter = ax.get_xaxis().get_major_formatter()
- with (pytest.warns(UserWarning, match='Attempting to set identical')
- if left == right else nullcontext()):
- ax.set_xlim(left, right)
- ax.get_xaxis()._update_ticks()
- assert formatter.offset == offset
- with (pytest.warns(UserWarning, match='Attempting to set identical')
- if left == right else nullcontext()):
- ax.set_xlim(right, left)
- ax.get_xaxis()._update_ticks()
- assert formatter.offset == offset
- @pytest.mark.parametrize('use_offset', use_offset_data)
- def test_use_offset(self, use_offset):
- with matplotlib.rc_context({'axes.formatter.useoffset': use_offset}):
- tmp_form = mticker.ScalarFormatter()
- assert use_offset == tmp_form.get_useOffset()
- @pytest.mark.parametrize(
- 'sci_type, scilimits, lim, orderOfMag, fewticks', scilimits_data)
- def test_scilimits(self, sci_type, scilimits, lim, orderOfMag, fewticks):
- tmp_form = mticker.ScalarFormatter()
- tmp_form.set_scientific(sci_type)
- tmp_form.set_powerlimits(scilimits)
- fig, ax = plt.subplots()
- ax.yaxis.set_major_formatter(tmp_form)
- ax.set_ylim(*lim)
- if fewticks:
- ax.yaxis.set_major_locator(mticker.MaxNLocator(4))
- tmp_form.set_locs(ax.yaxis.get_majorticklocs())
- assert orderOfMag == tmp_form.orderOfMagnitude
- class FakeAxis:
- """Allow Formatter to be called without having a "full" plot set up."""
- def __init__(self, vmin=1, vmax=10):
- self.vmin = vmin
- self.vmax = vmax
- def get_view_interval(self):
- return self.vmin, self.vmax
- class TestLogFormatterExponent:
- param_data = [
- (True, 4, np.arange(-3, 4.0), np.arange(-3, 4.0),
- ['-3', '-2', '-1', '0', '1', '2', '3']),
- # With labelOnlyBase=False, non-integer powers should be nicely
- # formatted.
- (False, 10, np.array([0.1, 0.00001, np.pi, 0.2, -0.2, -0.00001]),
- range(6), ['0.1', '1e-05', '3.14', '0.2', '-0.2', '-1e-05']),
- (False, 50, np.array([3, 5, 12, 42], dtype='float'), range(6),
- ['3', '5', '12', '42']),
- ]
- base_data = [2.0, 5.0, 10.0, np.pi, np.e]
- @pytest.mark.parametrize(
- 'labelOnlyBase, exponent, locs, positions, expected', param_data)
- @pytest.mark.parametrize('base', base_data)
- def test_basic(self, labelOnlyBase, base, exponent, locs, positions,
- expected):
- formatter = mticker.LogFormatterExponent(base=base,
- labelOnlyBase=labelOnlyBase)
- formatter.axis = FakeAxis(1, base**exponent)
- vals = base**locs
- labels = [formatter(x, pos) for (x, pos) in zip(vals, positions)]
- assert labels == expected
- def test_blank(self):
- # Should be a blank string for non-integer powers if labelOnlyBase=True
- formatter = mticker.LogFormatterExponent(base=10, labelOnlyBase=True)
- formatter.axis = FakeAxis()
- assert formatter(10**0.1) == ''
- class TestLogFormatterMathtext:
- fmt = mticker.LogFormatterMathtext()
- test_data = [
- (0, 1, '$\\mathdefault{10^{0}}$'),
- (0, 1e-2, '$\\mathdefault{10^{-2}}$'),
- (0, 1e2, '$\\mathdefault{10^{2}}$'),
- (3, 1, '$\\mathdefault{1}$'),
- (3, 1e-2, '$\\mathdefault{0.01}$'),
- (3, 1e2, '$\\mathdefault{100}$'),
- (3, 1e-3, '$\\mathdefault{10^{-3}}$'),
- (3, 1e3, '$\\mathdefault{10^{3}}$'),
- ]
- @pytest.mark.parametrize('min_exponent, value, expected', test_data)
- def test_min_exponent(self, min_exponent, value, expected):
- with matplotlib.rc_context({'axes.formatter.min_exponent':
- min_exponent}):
- assert self.fmt(value) == expected
- class TestLogFormatterSciNotation:
- test_data = [
- (2, 0.03125, '$\\mathdefault{2^{-5}}$'),
- (2, 1, '$\\mathdefault{2^{0}}$'),
- (2, 32, '$\\mathdefault{2^{5}}$'),
- (2, 0.0375, '$\\mathdefault{1.2\\times2^{-5}}$'),
- (2, 1.2, '$\\mathdefault{1.2\\times2^{0}}$'),
- (2, 38.4, '$\\mathdefault{1.2\\times2^{5}}$'),
- (10, -1, '$\\mathdefault{-10^{0}}$'),
- (10, 1e-05, '$\\mathdefault{10^{-5}}$'),
- (10, 1, '$\\mathdefault{10^{0}}$'),
- (10, 100000, '$\\mathdefault{10^{5}}$'),
- (10, 2e-05, '$\\mathdefault{2\\times10^{-5}}$'),
- (10, 2, '$\\mathdefault{2\\times10^{0}}$'),
- (10, 200000, '$\\mathdefault{2\\times10^{5}}$'),
- (10, 5e-05, '$\\mathdefault{5\\times10^{-5}}$'),
- (10, 5, '$\\mathdefault{5\\times10^{0}}$'),
- (10, 500000, '$\\mathdefault{5\\times10^{5}}$'),
- ]
- @pytest.mark.style('default')
- @pytest.mark.parametrize('base, value, expected', test_data)
- def test_basic(self, base, value, expected):
- formatter = mticker.LogFormatterSciNotation(base=base)
- formatter.sublabel = {1, 2, 5, 1.2}
- with matplotlib.rc_context({'text.usetex': False}):
- assert formatter(value) == expected
- class TestLogFormatter:
- pprint_data = [
- (3.141592654e-05, 0.001, '3.142e-5'),
- (0.0003141592654, 0.001, '3.142e-4'),
- (0.003141592654, 0.001, '3.142e-3'),
- (0.03141592654, 0.001, '3.142e-2'),
- (0.3141592654, 0.001, '3.142e-1'),
- (3.141592654, 0.001, '3.142'),
- (31.41592654, 0.001, '3.142e1'),
- (314.1592654, 0.001, '3.142e2'),
- (3141.592654, 0.001, '3.142e3'),
- (31415.92654, 0.001, '3.142e4'),
- (314159.2654, 0.001, '3.142e5'),
- (1e-05, 0.001, '1e-5'),
- (0.0001, 0.001, '1e-4'),
- (0.001, 0.001, '1e-3'),
- (0.01, 0.001, '1e-2'),
- (0.1, 0.001, '1e-1'),
- (1, 0.001, '1'),
- (10, 0.001, '10'),
- (100, 0.001, '100'),
- (1000, 0.001, '1000'),
- (10000, 0.001, '1e4'),
- (100000, 0.001, '1e5'),
- (3.141592654e-05, 0.015, '0'),
- (0.0003141592654, 0.015, '0'),
- (0.003141592654, 0.015, '0.003'),
- (0.03141592654, 0.015, '0.031'),
- (0.3141592654, 0.015, '0.314'),
- (3.141592654, 0.015, '3.142'),
- (31.41592654, 0.015, '31.416'),
- (314.1592654, 0.015, '314.159'),
- (3141.592654, 0.015, '3141.593'),
- (31415.92654, 0.015, '31415.927'),
- (314159.2654, 0.015, '314159.265'),
- (1e-05, 0.015, '0'),
- (0.0001, 0.015, '0'),
- (0.001, 0.015, '0.001'),
- (0.01, 0.015, '0.01'),
- (0.1, 0.015, '0.1'),
- (1, 0.015, '1'),
- (10, 0.015, '10'),
- (100, 0.015, '100'),
- (1000, 0.015, '1000'),
- (10000, 0.015, '10000'),
- (100000, 0.015, '100000'),
- (3.141592654e-05, 0.5, '0'),
- (0.0003141592654, 0.5, '0'),
- (0.003141592654, 0.5, '0.003'),
- (0.03141592654, 0.5, '0.031'),
- (0.3141592654, 0.5, '0.314'),
- (3.141592654, 0.5, '3.142'),
- (31.41592654, 0.5, '31.416'),
- (314.1592654, 0.5, '314.159'),
- (3141.592654, 0.5, '3141.593'),
- (31415.92654, 0.5, '31415.927'),
- (314159.2654, 0.5, '314159.265'),
- (1e-05, 0.5, '0'),
- (0.0001, 0.5, '0'),
- (0.001, 0.5, '0.001'),
- (0.01, 0.5, '0.01'),
- (0.1, 0.5, '0.1'),
- (1, 0.5, '1'),
- (10, 0.5, '10'),
- (100, 0.5, '100'),
- (1000, 0.5, '1000'),
- (10000, 0.5, '10000'),
- (100000, 0.5, '100000'),
- (3.141592654e-05, 5, '0'),
- (0.0003141592654, 5, '0'),
- (0.003141592654, 5, '0'),
- (0.03141592654, 5, '0.03'),
- (0.3141592654, 5, '0.31'),
- (3.141592654, 5, '3.14'),
- (31.41592654, 5, '31.42'),
- (314.1592654, 5, '314.16'),
- (3141.592654, 5, '3141.59'),
- (31415.92654, 5, '31415.93'),
- (314159.2654, 5, '314159.27'),
- (1e-05, 5, '0'),
- (0.0001, 5, '0'),
- (0.001, 5, '0'),
- (0.01, 5, '0.01'),
- (0.1, 5, '0.1'),
- (1, 5, '1'),
- (10, 5, '10'),
- (100, 5, '100'),
- (1000, 5, '1000'),
- (10000, 5, '10000'),
- (100000, 5, '100000'),
- (3.141592654e-05, 100, '0'),
- (0.0003141592654, 100, '0'),
- (0.003141592654, 100, '0'),
- (0.03141592654, 100, '0'),
- (0.3141592654, 100, '0.3'),
- (3.141592654, 100, '3.1'),
- (31.41592654, 100, '31.4'),
- (314.1592654, 100, '314.2'),
- (3141.592654, 100, '3141.6'),
- (31415.92654, 100, '31415.9'),
- (314159.2654, 100, '314159.3'),
- (1e-05, 100, '0'),
- (0.0001, 100, '0'),
- (0.001, 100, '0'),
- (0.01, 100, '0'),
- (0.1, 100, '0.1'),
- (1, 100, '1'),
- (10, 100, '10'),
- (100, 100, '100'),
- (1000, 100, '1000'),
- (10000, 100, '10000'),
- (100000, 100, '100000'),
- (3.141592654e-05, 1000000.0, '3.1e-5'),
- (0.0003141592654, 1000000.0, '3.1e-4'),
- (0.003141592654, 1000000.0, '3.1e-3'),
- (0.03141592654, 1000000.0, '3.1e-2'),
- (0.3141592654, 1000000.0, '3.1e-1'),
- (3.141592654, 1000000.0, '3.1'),
- (31.41592654, 1000000.0, '3.1e1'),
- (314.1592654, 1000000.0, '3.1e2'),
- (3141.592654, 1000000.0, '3.1e3'),
- (31415.92654, 1000000.0, '3.1e4'),
- (314159.2654, 1000000.0, '3.1e5'),
- (1e-05, 1000000.0, '1e-5'),
- (0.0001, 1000000.0, '1e-4'),
- (0.001, 1000000.0, '1e-3'),
- (0.01, 1000000.0, '1e-2'),
- (0.1, 1000000.0, '1e-1'),
- (1, 1000000.0, '1'),
- (10, 1000000.0, '10'),
- (100, 1000000.0, '100'),
- (1000, 1000000.0, '1000'),
- (10000, 1000000.0, '1e4'),
- (100000, 1000000.0, '1e5'),
- ]
- @pytest.mark.parametrize('value, domain, expected', pprint_data)
- def test_pprint(self, value, domain, expected):
- fmt = mticker.LogFormatter()
- label = fmt._pprint_val(value, domain)
- assert label == expected
- def _sub_labels(self, axis, subs=()):
- "Test whether locator marks subs to be labeled"
- fmt = axis.get_minor_formatter()
- minor_tlocs = axis.get_minorticklocs()
- fmt.set_locs(minor_tlocs)
- coefs = minor_tlocs / 10**(np.floor(np.log10(minor_tlocs)))
- label_expected = [round(c) in subs for c in coefs]
- label_test = [fmt(x) != '' for x in minor_tlocs]
- assert label_test == label_expected
- @pytest.mark.style('default')
- def test_sublabel(self):
- # test label locator
- fig, ax = plt.subplots()
- ax.set_xscale('log')
- ax.xaxis.set_major_locator(mticker.LogLocator(base=10, subs=[]))
- ax.xaxis.set_minor_locator(mticker.LogLocator(base=10,
- subs=np.arange(2, 10)))
- ax.xaxis.set_major_formatter(mticker.LogFormatter(labelOnlyBase=True))
- ax.xaxis.set_minor_formatter(mticker.LogFormatter(labelOnlyBase=False))
- # axis range above 3 decades, only bases are labeled
- ax.set_xlim(1, 1e4)
- fmt = ax.xaxis.get_major_formatter()
- fmt.set_locs(ax.xaxis.get_majorticklocs())
- show_major_labels = [fmt(x) != ''
- for x in ax.xaxis.get_majorticklocs()]
- assert np.all(show_major_labels)
- self._sub_labels(ax.xaxis, subs=[])
- # For the next two, if the numdec threshold in LogFormatter.set_locs
- # were 3, then the label sub would be 3 for 2-3 decades and (2, 5)
- # for 1-2 decades. With a threshold of 1, subs are not labeled.
- # axis range at 2 to 3 decades
- ax.set_xlim(1, 800)
- self._sub_labels(ax.xaxis, subs=[])
- # axis range at 1 to 2 decades
- ax.set_xlim(1, 80)
- self._sub_labels(ax.xaxis, subs=[])
- # axis range at 0.4 to 1 decades, label subs 2, 3, 4, 6
- ax.set_xlim(1, 8)
- self._sub_labels(ax.xaxis, subs=[2, 3, 4, 6])
- # axis range at 0 to 0.4 decades, label all
- ax.set_xlim(0.5, 0.9)
- self._sub_labels(ax.xaxis, subs=np.arange(2, 10, dtype=int))
- @pytest.mark.parametrize('val', [1, 10, 100, 1000])
- def test_LogFormatter_call(self, val):
- # test _num_to_string method used in __call__
- temp_lf = mticker.LogFormatter()
- temp_lf.axis = FakeAxis()
- assert temp_lf(val) == str(val)
- class TestLogitFormatter:
- @staticmethod
- def logit_deformatter(string):
- r"""
- Parser to convert string as r'$\mathdefault{1.41\cdot10^{-4}}$' in
- float 1.41e-4, as '0.5' or as r'$\mathdefault{\frac{1}{2}}$' in float
- 0.5,
- """
- match = re.match(
- r"[^\d]*"
- r"(?P<comp>1-)?"
- r"(?P<mant>\d*\.?\d*)?"
- r"(?:\\cdot)?"
- r"(?:10\^\{(?P<expo>-?\d*)})?"
- r"[^\d]*$",
- string,
- )
- if match:
- comp = match["comp"] is not None
- mantissa = float(match["mant"]) if match["mant"] else 1
- expo = int(match["expo"]) if match["expo"] is not None else 0
- value = mantissa * 10 ** expo
- if match["mant"] or match["expo"] is not None:
- if comp:
- return 1 - value
- return value
- match = re.match(
- r"[^\d]*\\frac\{(?P<num>\d+)\}\{(?P<deno>\d+)\}[^\d]*$", string
- )
- if match:
- num, deno = float(match["num"]), float(match["deno"])
- return num / deno
- raise ValueError("not formatted by LogitFormatter")
- @pytest.mark.parametrize(
- "fx, x",
- [
- (r"STUFF0.41OTHERSTUFF", 0.41),
- (r"STUFF1.41\cdot10^{-2}OTHERSTUFF", 1.41e-2),
- (r"STUFF1-0.41OTHERSTUFF", 1 - 0.41),
- (r"STUFF1-1.41\cdot10^{-2}OTHERSTUFF", 1 - 1.41e-2),
- (r"STUFF", None),
- (r"STUFF12.4e-3OTHERSTUFF", None),
- ],
- )
- def test_logit_deformater(self, fx, x):
- if x is None:
- with pytest.raises(ValueError):
- TestLogitFormatter.logit_deformatter(fx)
- else:
- y = TestLogitFormatter.logit_deformatter(fx)
- assert _LogitHelper.isclose(x, y)
- decade_test = sorted(
- [10 ** (-i) for i in range(1, 10)]
- + [1 - 10 ** (-i) for i in range(1, 10)]
- + [1 / 2]
- )
- @pytest.mark.parametrize("x", decade_test)
- def test_basic(self, x):
- """
- Test the formatted value correspond to the value for ideal ticks in
- logit space.
- """
- formatter = mticker.LogitFormatter(use_overline=False)
- formatter.set_locs(self.decade_test)
- s = formatter(x)
- x2 = TestLogitFormatter.logit_deformatter(s)
- assert _LogitHelper.isclose(x, x2)
- @pytest.mark.parametrize("x", (-1, -0.5, -0.1, 1.1, 1.5, 2))
- def test_invalid(self, x):
- """
- Test that invalid value are formatted with empty string without
- raising exception.
- """
- formatter = mticker.LogitFormatter(use_overline=False)
- formatter.set_locs(self.decade_test)
- s = formatter(x)
- assert s == ""
- @pytest.mark.parametrize("x", 1 / (1 + np.exp(-np.linspace(-7, 7, 10))))
- def test_variablelength(self, x):
- """
- The format length should change depending on the neighbor labels.
- """
- formatter = mticker.LogitFormatter(use_overline=False)
- for N in (10, 20, 50, 100, 200, 1000, 2000, 5000, 10000):
- if x + 1 / N < 1:
- formatter.set_locs([x - 1 / N, x, x + 1 / N])
- sx = formatter(x)
- sx1 = formatter(x + 1 / N)
- d = (
- TestLogitFormatter.logit_deformatter(sx1)
- - TestLogitFormatter.logit_deformatter(sx)
- )
- assert 0 < d < 2 / N
- lims_minor_major = [
- (True, (5e-8, 1 - 5e-8), ((25, False), (75, False))),
- (True, (5e-5, 1 - 5e-5), ((25, False), (75, True))),
- (True, (5e-2, 1 - 5e-2), ((25, True), (75, True))),
- (False, (0.75, 0.76, 0.77), ((7, True), (25, True), (75, True))),
- ]
- @pytest.mark.parametrize("method, lims, cases", lims_minor_major)
- def test_minor_vs_major(self, method, lims, cases):
- """
- Test minor/major displays.
- """
- if method:
- min_loc = mticker.LogitLocator(minor=True)
- ticks = min_loc.tick_values(*lims)
- else:
- ticks = np.array(lims)
- min_form = mticker.LogitFormatter(minor=True)
- for threshold, has_minor in cases:
- min_form.set_minor_threshold(threshold)
- formatted = min_form.format_ticks(ticks)
- labelled = [f for f in formatted if len(f) > 0]
- if has_minor:
- assert len(labelled) > 0, (threshold, has_minor)
- else:
- assert len(labelled) == 0, (threshold, has_minor)
- def test_minor_number(self):
- """
- Test the parameter minor_number
- """
- min_loc = mticker.LogitLocator(minor=True)
- min_form = mticker.LogitFormatter(minor=True)
- ticks = min_loc.tick_values(5e-2, 1 - 5e-2)
- for minor_number in (2, 4, 8, 16):
- min_form.set_minor_number(minor_number)
- formatted = min_form.format_ticks(ticks)
- labelled = [f for f in formatted if len(f) > 0]
- assert len(labelled) == minor_number
- def test_use_overline(self):
- """
- Test the parameter use_overline
- """
- x = 1 - 1e-2
- fx1 = r"$\mathdefault{1-10^{-2}}$"
- fx2 = r"$\mathdefault{\overline{10^{-2}}}$"
- form = mticker.LogitFormatter(use_overline=False)
- assert form(x) == fx1
- form.use_overline(True)
- assert form(x) == fx2
- form.use_overline(False)
- assert form(x) == fx1
- def test_one_half(self):
- """
- Test the parameter one_half
- """
- form = mticker.LogitFormatter()
- assert r"\frac{1}{2}" in form(1/2)
- form.set_one_half("1/2")
- assert "1/2" in form(1/2)
- form.set_one_half("one half")
- assert "one half" in form(1/2)
- @pytest.mark.parametrize("N", (100, 253, 754))
- def test_format_data_short(self, N):
- locs = np.linspace(0, 1, N)[1:-1]
- form = mticker.LogitFormatter()
- for x in locs:
- fx = form.format_data_short(x)
- if fx.startswith("1-"):
- x2 = 1 - float(fx[2:])
- else:
- x2 = float(fx)
- assert np.abs(x - x2) < 1 / N
- class TestFormatStrFormatter:
- def test_basic(self):
- # test % style formatter
- tmp_form = mticker.FormatStrFormatter('%05d')
- assert '00002' == tmp_form(2)
- class TestStrMethodFormatter:
- test_data = [
- ('{x:05d}', (2,), '00002'),
- ('{x:03d}-{pos:02d}', (2, 1), '002-01'),
- ]
- @pytest.mark.parametrize('format, input, expected', test_data)
- def test_basic(self, format, input, expected):
- fmt = mticker.StrMethodFormatter(format)
- assert fmt(*input) == expected
- class TestEngFormatter:
- # (unicode_minus, input, expected) where ''expected'' corresponds to the
- # outputs respectively returned when (places=None, places=0, places=2)
- # unicode_minus is a boolean value for the rcParam['axes.unicode_minus']
- raw_format_data = [
- (False, -1234.56789, ('-1.23457 k', '-1 k', '-1.23 k')),
- (True, -1234.56789, ('\N{MINUS SIGN}1.23457 k', '\N{MINUS SIGN}1 k',
- '\N{MINUS SIGN}1.23 k')),
- (False, -1.23456789, ('-1.23457', '-1', '-1.23')),
- (True, -1.23456789, ('\N{MINUS SIGN}1.23457', '\N{MINUS SIGN}1',
- '\N{MINUS SIGN}1.23')),
- (False, -0.123456789, ('-123.457 m', '-123 m', '-123.46 m')),
- (True, -0.123456789, ('\N{MINUS SIGN}123.457 m', '\N{MINUS SIGN}123 m',
- '\N{MINUS SIGN}123.46 m')),
- (False, -0.00123456789, ('-1.23457 m', '-1 m', '-1.23 m')),
- (True, -0.00123456789, ('\N{MINUS SIGN}1.23457 m', '\N{MINUS SIGN}1 m',
- '\N{MINUS SIGN}1.23 m')),
- (True, -0.0, ('0', '0', '0.00')),
- (True, -0, ('0', '0', '0.00')),
- (True, 0, ('0', '0', '0.00')),
- (True, 1.23456789e-6, ('1.23457 µ', '1 µ', '1.23 µ')),
- (True, 0.123456789, ('123.457 m', '123 m', '123.46 m')),
- (True, 0.1, ('100 m', '100 m', '100.00 m')),
- (True, 1, ('1', '1', '1.00')),
- (True, 1.23456789, ('1.23457', '1', '1.23')),
- # places=0: corner-case rounding
- (True, 999.9, ('999.9', '1 k', '999.90')),
- # corner-case rounding for all
- (True, 999.9999, ('1 k', '1 k', '1.00 k')),
- # negative corner-case
- (False, -999.9999, ('-1 k', '-1 k', '-1.00 k')),
- (True, -999.9999, ('\N{MINUS SIGN}1 k', '\N{MINUS SIGN}1 k',
- '\N{MINUS SIGN}1.00 k')),
- (True, 1000, ('1 k', '1 k', '1.00 k')),
- (True, 1001, ('1.001 k', '1 k', '1.00 k')),
- (True, 100001, ('100.001 k', '100 k', '100.00 k')),
- (True, 987654.321, ('987.654 k', '988 k', '987.65 k')),
- # OoR value (> 1000 Y)
- (True, 1.23e27, ('1230 Y', '1230 Y', '1230.00 Y'))
- ]
- @pytest.mark.parametrize('unicode_minus, input, expected', raw_format_data)
- def test_params(self, unicode_minus, input, expected):
- """
- Test the formatting of EngFormatter for various values of the 'places'
- argument, in several cases:
- 0. without a unit symbol but with a (default) space separator;
- 1. with both a unit symbol and a (default) space separator;
- 2. with both a unit symbol and some non default separators;
- 3. without a unit symbol but with some non default separators.
- Note that cases 2. and 3. are looped over several separator strings.
- """
- plt.rcParams['axes.unicode_minus'] = unicode_minus
- UNIT = 's' # seconds
- DIGITS = '0123456789' # %timeit showed 10-20% faster search than set
- # Case 0: unit='' (default) and sep=' ' (default).
- # 'expected' already corresponds to this reference case.
- exp_outputs = expected
- formatters = (
- mticker.EngFormatter(), # places=None (default)
- mticker.EngFormatter(places=0),
- mticker.EngFormatter(places=2)
- )
- for _formatter, _exp_output in zip(formatters, exp_outputs):
- assert _formatter(input) == _exp_output
- # Case 1: unit=UNIT and sep=' ' (default).
- # Append a unit symbol to the reference case.
- # Beware of the values in [1, 1000), where there is no prefix!
- exp_outputs = (_s + " " + UNIT if _s[-1] in DIGITS # case w/o prefix
- else _s + UNIT for _s in expected)
- formatters = (
- mticker.EngFormatter(unit=UNIT), # places=None (default)
- mticker.EngFormatter(unit=UNIT, places=0),
- mticker.EngFormatter(unit=UNIT, places=2)
- )
- for _formatter, _exp_output in zip(formatters, exp_outputs):
- assert _formatter(input) == _exp_output
- # Test several non default separators: no separator, a narrow
- # no-break space (unicode character) and an extravagant string.
- for _sep in ("", "\N{NARROW NO-BREAK SPACE}", "@_@"):
- # Case 2: unit=UNIT and sep=_sep.
- # Replace the default space separator from the reference case
- # with the tested one `_sep` and append a unit symbol to it.
- exp_outputs = (_s + _sep + UNIT if _s[-1] in DIGITS # no prefix
- else _s.replace(" ", _sep) + UNIT
- for _s in expected)
- formatters = (
- mticker.EngFormatter(unit=UNIT, sep=_sep), # places=None
- mticker.EngFormatter(unit=UNIT, places=0, sep=_sep),
- mticker.EngFormatter(unit=UNIT, places=2, sep=_sep)
- )
- for _formatter, _exp_output in zip(formatters, exp_outputs):
- assert _formatter(input) == _exp_output
- # Case 3: unit='' (default) and sep=_sep.
- # Replace the default space separator from the reference case
- # with the tested one `_sep`. Reference case is already unitless.
- exp_outputs = (_s.replace(" ", _sep) for _s in expected)
- formatters = (
- mticker.EngFormatter(sep=_sep), # places=None (default)
- mticker.EngFormatter(places=0, sep=_sep),
- mticker.EngFormatter(places=2, sep=_sep)
- )
- for _formatter, _exp_output in zip(formatters, exp_outputs):
- assert _formatter(input) == _exp_output
- def test_engformatter_usetex_useMathText():
- fig, ax = plt.subplots()
- ax.plot([0, 500, 1000], [0, 500, 1000])
- ax.set_xticks([0, 500, 1000])
- for formatter in (mticker.EngFormatter(usetex=True),
- mticker.EngFormatter(useMathText=True)):
- ax.xaxis.set_major_formatter(formatter)
- fig.canvas.draw()
- x_tick_label_text = [labl.get_text() for labl in ax.get_xticklabels()]
- # Checking if the dollar `$` signs have been inserted around numbers
- # in tick labels.
- assert x_tick_label_text == ['$0$', '$500$', '$1$ k']
- class TestPercentFormatter:
- percent_data = [
- # Check explicitly set decimals over different intervals and values
- (100, 0, '%', 120, 100, '120%'),
- (100, 0, '%', 100, 90, '100%'),
- (100, 0, '%', 90, 50, '90%'),
- (100, 0, '%', -1.7, 40, '-2%'),
- (100, 1, '%', 90.0, 100, '90.0%'),
- (100, 1, '%', 80.1, 90, '80.1%'),
- (100, 1, '%', 70.23, 50, '70.2%'),
- # 60.554 instead of 60.55: see https://bugs.python.org/issue5118
- (100, 1, '%', -60.554, 40, '-60.6%'),
- # Check auto decimals over different intervals and values
- (100, None, '%', 95, 1, '95.00%'),
- (1.0, None, '%', 3, 6, '300%'),
- (17.0, None, '%', 1, 8.5, '6%'),
- (17.0, None, '%', 1, 8.4, '5.9%'),
- (5, None, '%', -100, 0.000001, '-2000.00000%'),
- # Check percent symbol
- (1.0, 2, None, 1.2, 100, '120.00'),
- (75, 3, '', 50, 100, '66.667'),
- (42, None, '^^Foobar$$', 21, 12, '50.0^^Foobar$$'),
- ]
- percent_ids = [
- # Check explicitly set decimals over different intervals and values
- 'decimals=0, x>100%',
- 'decimals=0, x=100%',
- 'decimals=0, x<100%',
- 'decimals=0, x<0%',
- 'decimals=1, x>100%',
- 'decimals=1, x=100%',
- 'decimals=1, x<100%',
- 'decimals=1, x<0%',
- # Check auto decimals over different intervals and values
- 'autodecimal, x<100%, display_range=1',
- 'autodecimal, x>100%, display_range=6 (custom xmax test)',
- 'autodecimal, x<100%, display_range=8.5 (autodecimal test 1)',
- 'autodecimal, x<100%, display_range=8.4 (autodecimal test 2)',
- 'autodecimal, x<-100%, display_range=1e-6 (tiny display range)',
- # Check percent symbol
- 'None as percent symbol',
- 'Empty percent symbol',
- 'Custom percent symbol',
- ]
- latex_data = [
- (False, False, r'50\{t}%'),
- (False, True, r'50\\\{t\}\%'),
- (True, False, r'50\{t}%'),
- (True, True, r'50\{t}%'),
- ]
- @pytest.mark.parametrize(
- 'xmax, decimals, symbol, x, display_range, expected',
- percent_data, ids=percent_ids)
- def test_basic(self, xmax, decimals, symbol,
- x, display_range, expected):
- formatter = mticker.PercentFormatter(xmax, decimals, symbol)
- with matplotlib.rc_context(rc={'text.usetex': False}):
- assert formatter.format_pct(x, display_range) == expected
- @pytest.mark.parametrize('is_latex, usetex, expected', latex_data)
- def test_latex(self, is_latex, usetex, expected):
- fmt = mticker.PercentFormatter(symbol='\\{t}%', is_latex=is_latex)
- with matplotlib.rc_context(rc={'text.usetex': usetex}):
- assert fmt.format_pct(50, 100) == expected
- def test_majformatter_type():
- fig, ax = plt.subplots()
- with pytest.raises(TypeError):
- ax.xaxis.set_major_formatter(matplotlib.ticker.LogLocator())
- def test_minformatter_type():
- fig, ax = plt.subplots()
- with pytest.raises(TypeError):
- ax.xaxis.set_minor_formatter(matplotlib.ticker.LogLocator())
- def test_majlocator_type():
- fig, ax = plt.subplots()
- with pytest.raises(TypeError):
- ax.xaxis.set_major_locator(matplotlib.ticker.LogFormatter())
- def test_minlocator_type():
- fig, ax = plt.subplots()
- with pytest.raises(TypeError):
- ax.xaxis.set_minor_locator(matplotlib.ticker.LogFormatter())
- def test_minorticks_rc():
- fig = plt.figure()
- def minorticksubplot(xminor, yminor, i):
- rc = {'xtick.minor.visible': xminor,
- 'ytick.minor.visible': yminor}
- with plt.rc_context(rc=rc):
- ax = fig.add_subplot(2, 2, i)
- assert (len(ax.xaxis.get_minor_ticks()) > 0) == xminor
- assert (len(ax.yaxis.get_minor_ticks()) > 0) == yminor
- minorticksubplot(False, False, 1)
- minorticksubplot(True, False, 2)
- minorticksubplot(False, True, 3)
- minorticksubplot(True, True, 4)
- @pytest.mark.parametrize('remove_overlapping_locs, expected_num',
- ((True, 6),
- (None, 6), # this tests the default
- (False, 9)))
- def test_remove_overlap(remove_overlapping_locs, expected_num):
- import numpy as np
- import matplotlib.dates as mdates
- t = np.arange("2018-11-03", "2018-11-06", dtype="datetime64")
- x = np.ones(len(t))
- fig, ax = plt.subplots()
- ax.plot(t, x)
- ax.xaxis.set_major_locator(mdates.DayLocator())
- ax.xaxis.set_major_formatter(mdates.DateFormatter('\n%a'))
- ax.xaxis.set_minor_locator(mdates.HourLocator((0, 6, 12, 18)))
- ax.xaxis.set_minor_formatter(mdates.DateFormatter('%H:%M'))
- # force there to be extra ticks
- ax.xaxis.get_minor_ticks(15)
- if remove_overlapping_locs is not None:
- ax.xaxis.remove_overlapping_locs = remove_overlapping_locs
- # check that getter/setter exists
- current = ax.xaxis.remove_overlapping_locs
- assert (current == ax.xaxis.get_remove_overlapping_locs())
- plt.setp(ax.xaxis, remove_overlapping_locs=current)
- new = ax.xaxis.remove_overlapping_locs
- assert (new == ax.xaxis.remove_overlapping_locs)
- # check that the accessors filter correctly
- # this is the method that does the actual filtering
- assert len(ax.xaxis.get_minorticklocs()) == expected_num
- # these three are derivative
- assert len(ax.xaxis.get_minor_ticks()) == expected_num
- assert len(ax.xaxis.get_minorticklabels()) == expected_num
- assert len(ax.xaxis.get_minorticklines()) == expected_num*2
- @pytest.mark.parametrize('sub', [
- ['hi', 'aardvark'],
- np.zeros((2, 2))])
- def test_bad_locator_subs(sub):
- ll = mticker.LogLocator()
- with pytest.raises(ValueError):
- ll.subs(sub)
|