1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
|
import os
import numpy as np
import pytest
from ase.lattice.cubic import FaceCenteredCubic
from ase.utils.plotting import SimplePlottingAxes
from ase.visualize.plot import plot_atoms
def test_matplotlib_plot(plt):
slab = FaceCenteredCubic('Au', size=(2, 2, 2))
fig, ax = plt.subplots()
plot_atoms(slab, ax, radii=0.5, rotation=('10x,10y,10z'),
show_unit_cell=0)
assert len(ax.patches) == len(slab)
print(ax)
class TestPlotManager:
@pytest.fixture
def xy_data(self):
return ([1, 2], [3, 4])
def test_plot_manager_error(self, figure):
with pytest.raises(AssertionError):
with SimplePlottingAxes(ax=None, show=False, filename=None):
raise AssertionError()
def test_plot_manager_no_file(self, xy_data, figure):
x, y = xy_data
with SimplePlottingAxes(ax=None, show=False, filename=None) as ax:
ax.plot(x, y)
assert np.allclose(ax.lines[0].get_xydata().transpose(), xy_data)
def test_plot_manager_axis_file(self, xy_data, figure):
filename = 'plot.png'
x, y = xy_data
ax = figure.add_subplot(111)
with SimplePlottingAxes(ax=ax, show=False,
filename=filename) as return_ax:
assert return_ax is ax
ax.plot(x, y)
assert os.path.isfile(filename)
|
