def test_plot_3(self):
matplotlib_scalar(self.grid,
pyplot.gca(), (0.1, 0.1, 0.1),
'z',
show_cell=True)
l = list(i for i in pyplot.gca().get_children()
if isinstance(i, Line2D))
assert len(l) == 12
def test_plot(self):
im = matplotlib_scalar(self.grid, pyplot.gca(), (0.1, 0.1, 0.1), 'z')
axes = pyplot.gcf().axes
assert len(axes) == 1
axes = axes[0]
ai = axes.findobj(AxesImage)
assert len(ai) == 1
ai = ai[0]
assert ai == im
testing.assert_allclose(axes.get_xlim(), (-0.55, 0.95))
testing.assert_allclose(axes.get_ylim(),
(-0.1 * 3.**.5 / 2, 0.9 * 3.**.5 / 2))
def test_plot_2(self):
expected_shape = (17, 10)
def pp(x):
testing.assert_equal(x.shape, expected_shape)
return x
im = matplotlib_scalar(self.grid,
pyplot.gca(), (0.1, 0.1, 0.1),
'z',
ppu=10,
margins=0,
postproc=pp)
pyplot.colorbar(im)
testing.assert_equal(im.get_size(), expected_shape[::-1])
def test_plot(self):
im = matplotlib_scalar(self.grid, pyplot.gca(), (0.1,0.1,0.1), 'z')
axes = pyplot.gcf().axes
assert len(axes) == 1
axes = axes[0]
ai = axes.findobj(AxesImage)
assert len(ai) == 1
ai = ai[0]
assert ai == im
testing.assert_allclose(axes.get_xlim(), (-0.55, 0.95))
testing.assert_allclose(axes.get_ylim(), (-0.1*3.**.5/2, 0.9*3.**.5/2))
from dfttools.types import Basis, Grid
from dfttools import presentation
from numericalunits import angstrom
from matplotlib import pyplot
import numpy
grid = Grid(
Basis((1*angstrom,1*angstrom,1*angstrom,0,0,-0.5), kind = 'triclinic'),
(
numpy.linspace(0,1,30,endpoint = False),
numpy.linspace(0,1,30,endpoint = False),
numpy.linspace(0,1,30,endpoint = False),
),
numpy.zeros((30,30,30)),
)
grid.values = numpy.prod(numpy.sin(grid.explicit_coordinates()*2*numpy.pi), axis = -1)
presentation.matplotlib_scalar(grid, pyplot.gca(), (0.1,0.1,0.1), 'z', show_cell = True)
pyplot.show()
def test_plot_error_1(self):
with self.assertRaises(ValueError):
matplotlib_scalar(self.wrong_dims, pyplot.gca(), (0.1, 0.1, 0.1),
'z')
from dfttools.types import Basis, Grid
from dfttools import presentation
from numericalunits import angstrom
from matplotlib import pyplot
import numpy
grid = Grid(
Basis((1 * angstrom, 1 * angstrom, 1 * angstrom, 0, 0, -0.5),
kind='triclinic'),
(
numpy.linspace(0, 1, 30, endpoint=False),
numpy.linspace(0, 1, 30, endpoint=False),
numpy.linspace(0, 1, 30, endpoint=False),
),
numpy.zeros((30, 30, 30)),
)
grid.values = numpy.prod(numpy.sin(grid.explicit_coordinates() * 2 * numpy.pi),
axis=-1)
presentation.matplotlib_scalar(grid,
pyplot.gca(), (0.1, 0.1, 0.1),
'z',
show_cell=True)
pyplot.show()
def test_plot_error_1(self):
with self.assertRaises(TypeError):
matplotlib_scalar(self.wrong_dims, pyplot.gca(), (0.1,0.1,0.1), 'z')
def test_plot_3(self):
im = matplotlib_scalar(self.grid, pyplot.gca(), (0.1,0.1,0.1), 'z', show_cell = True)
l = list(i for i in pyplot.gca().get_children() if isinstance(i, Line2D))
print len(l)
assert len(l) == 12
def test_plot_2(self):
im = matplotlib_scalar(self.grid, pyplot.gca(), (0.1,0.1,0.1), 'z', ppu = 10)
testing.assert_equal(im.get_size(), (round(10*3.**.5/2), 15))
