import numpy as np

import pytest

import matplotlib.pyplot as plt

from matplotlib.spines import Spines

from matplotlib.testing.decorators import check_figures_equal, image_comparison

def test_spine_class():

"""Test Spines and SpinesProxy in isolation."""

class SpineMock:

def __init__(self):

self.val = None

def set_val(self, val):

self.val = val

spines_dict = {

'left': SpineMock(),

'right': SpineMock(),

'top': SpineMock(),

'bottom': SpineMock(),

}

spines = Spines(**spines_dict)

assert spines['left'] is spines_dict['left']

assert spines.left is spines_dict['left']

spines[['left', 'right']].set_val('x')

assert spines.left.val == 'x'

assert spines.right.val == 'x'

assert spines.top.val is None

assert spines.bottom.val is None

spines[:].set_val('y')

assert all(spine.val == 'y' for spine in spines.values())

with pytest.raises(KeyError, match='foo'):

spines['foo']

with pytest.raises(KeyError, match='foo, bar'):

spines[['left', 'foo', 'right', 'bar']]

with pytest.raises(ValueError, match='single list'):

spines['left', 'right']

with pytest.raises(ValueError, match='Spines does not support slicing'):

spines['left':'right']

with pytest.raises(ValueError, match='Spines does not support slicing'):

spines['top':]

@image_comparison(['spines_axes_positions'])

def test_spines_axes_positions():

# SF bug 2852168

fig = plt.figure()

x = np.linspace(0, 2*np.pi, 100)

y = 2*np.sin(x)

ax = fig.add_subplot(1, 1, 1)

ax.set_title('centered spines')

ax.plot(x, y)

ax.spines.right.set_position(('axes', 0.1))

ax.yaxis.set_ticks_position('right')

ax.spines.top.set_position(('axes', 0.25))

ax.xaxis.set_ticks_position('top')

ax.spines.left.set_color('none')

ax.spines.bottom.set_color('none')

@image_comparison(['spines_data_positions'])

def test_spines_data_positions():

fig, ax = plt.subplots()

ax.spines.left.set_position(('data', -1.5))

ax.spines.top.set_position(('data', 0.5))

ax.spines.right.set_position(('data', -0.5))

ax.spines.bottom.set_position('zero')

ax.set_xlim([-2, 2])

ax.set_ylim([-2, 2])

@check_figures_equal(extensions=["png"])

def test_spine_nonlinear_data_positions(fig_test, fig_ref):

plt.style.use("default")

ax = fig_test.add_subplot()

ax.set(xscale="log", xlim=(.1, 1))

# Use position="data" to visually swap the left and right spines, using

# linewidth to distinguish them. The calls to tick_params removes labels

# (for image comparison purposes) and harmonizes tick positions with the

# reference).

ax.spines.left.set_position(("data", 1))

ax.spines.left.set_linewidth(2)

ax.spines.right.set_position(("data", .1))

ax.tick_params(axis="y", labelleft=False, direction="in")

ax = fig_ref.add_subplot()

ax.set(xscale="log", xlim=(.1, 1))

ax.spines.right.set_linewidth(2)

ax.tick_params(axis="y", labelleft=False, left=False, right=True)

@image_comparison(['spines_capstyle'])

def test_spines_capstyle():

# issue 2542

plt.rc('axes', linewidth=20)

fig, ax = plt.subplots()

ax.set_xticks([])

ax.set_yticks([])

def test_label_without_ticks():

fig, ax = plt.subplots()

plt.subplots_adjust(left=0.3, bottom=0.3)

ax.plot(np.arange(10))

ax.yaxis.set_ticks_position('left')

ax.spines.left.set_position(('outward', 30))

ax.spines.right.set_visible(False)

ax.set_ylabel('y label')

ax.xaxis.set_ticks_position('bottom')

ax.spines.bottom.set_position(('outward', 30))

ax.spines.top.set_visible(False)

ax.set_xlabel('x label')

ax.xaxis.set_ticks([])

ax.yaxis.set_ticks([])

plt.draw()

spine = ax.spines.left

spinebbox = spine.get_transform().transform_path(

spine.get_path()).get_extents()

assert ax.yaxis.label.get_position()[0] < spinebbox.xmin, \

"Y-Axis label not left of the spine"

spine = ax.spines.bottom

spinebbox = spine.get_transform().transform_path(

spine.get_path()).get_extents()

assert ax.xaxis.label.get_position()[1] < spinebbox.ymin, \

"X-Axis label not below the spine"