def test_invalidkeywords():
with pytest.raises(TypeError):
display.set(nonsense=5.0)
with pytest.raises(ValueError):
display.get("nonsense")
def test_invalidkeywords():
with pytest.raises(TypeError):
display.set(nonsense=5.0)
with pytest.raises(ValueError):
display.get("nonsense")
def test_dof():
"""
dof, dof_fnumber, dof_focaldist
"""
display.set(dof=True, dof_fnumber=3.2, dof_focaldist=12.0)
assert display.get("dof")
assert display.get("dof_fnumber") == pytest.approx(3.2)
assert display.get("dof_focaldist") == 12.0
def test_dof():
"""
dof, dof_fnumber, dof_focaldist
"""
display.set(dof=True, dof_fnumber=3.2, dof_focaldist=12.0)
assert display.get("dof")
assert display.get("dof_fnumber") == pytest.approx(3.2)
assert display.get("dof_focaldist") == 12.0
def test_size():
"""
size
"""
currsize = display.get("size")
assert len(currsize) == 2
display.set(size=currsize[::1])
assert display.get("size") == currsize[::1]
display.set(size=(100,200))
assert display.get("size") == [100, 200]
def test_size():
"""
size
"""
currsize = display.get("size")
assert len(currsize) == 2
display.set(size=currsize[::1])
assert display.get("size") == currsize[::1]
display.set(size=(100, 200))
assert display.get("size") == [100, 200]
def test_snapshot(tmpdir, file_3nob):
from vmd import molecule, molrep, display
m = molecule.load("mae", file_3nob)
molrep.addrep(m, "NewCartoon")
display.set(size=(512, 512))
tmpdir = str(tmpdir)
render.render("snapshot", os.path.join(tmpdir, "test.tga"))
# If it's an empty header, the file size will be way too small here
assert os.path.getsize(os.path.join(tmpdir, "test.tga")) == 786450
def init_display(display_options, axes_options):
"""Function for initializing VMD scene options
Parameters
----------
display_options : dict
scene/display options for VMD. These are specified as key-value
pairs for each attribute in the VMD display options list. For
this complete list, see the following:
https://www.ks.uiuc.edu/Research/vmd/current/ug/node126.html
"""
display.set(**display_options)
if 'set_location' in axes_options.keys():
axes.set_location(axes_options['set_location'])
display.update()
def test_rendergetset():
"""
tests:
antialias, ambientocclusion, aoambient, aodirect, shadows,
"""
display.set(antialias=False, ambientocclusion=True)
with pytest.raises(TypeError):
display.set(antialias=5)
assert not display.get("antialias")
assert display.get("ambientocclusion")
display.set(ambientocclusion=False, aoambient=1.0, aodirect=3.0)
assert not display.get("ambientocclusion")
assert display.get("aoambient") == 1.0
assert display.get("aodirect") == 3.0
display.set(shadows=True)
assert display.get("shadows")
def test_rendergetset():
"""
tests:
antialias, ambientocclusion, aoambient, aodirect, shadows,
"""
display.set(antialias=False, ambientocclusion=True)
with pytest.raises(TypeError):
display.set(antialias=5)
assert not display.get("antialias")
assert display.get("ambientocclusion")
display.set(ambientocclusion=False, aoambient=1.0, aodirect=3.0)
assert not display.get("ambientocclusion")
assert display.get("aoambient") == 1.0
assert display.get("aodirect") == 3.0
display.set(shadows=True)
assert display.get("shadows")
def test_3dgetset():
"""
tests: eyesep, focallength, height, distance, nearclip, farclip,
depthcueue, culling, stereo, projection
"""
modes = display.stereomodes()
assert len(modes) == 11
display.set(eyesep=2.0, focallength=10.0, height=200,
distance=20, stereo=modes[0])
assert display.get("eyesep") == 2.0
assert display.get("focallength") == 10.0
assert display.get("height") == 200
assert display.get("distance") == 20
assert display.get("stereo") == modes[0]
# Default values from DisplayDevice.C
assert display.get("nearclip") == 0.5
assert display.get("farclip") == 10.0
display.set(nearclip=10.0, farclip=30.0)
assert display.get("nearclip") == 10.0
assert display.get("farclip") == 30.0
with pytest.raises(ValueError):
display.set(nearclip=10.0, farclip=5.0)
display.set(culling=False)
assert not display.get("culling")
with pytest.raises(TypeError):
display.set(culling=5.0)
assert "QuadBuffered" in modes
display.set(stereo="QuadBuffered")
assert display.get("stereo") == "QuadBuffered"
display.set(projection=display.PROJ_PERSP)
assert display.get("projection") == display.PROJ_PERSP
with pytest.raises(ValueError):
display.set(projection="invalid")
display.set(depthcue=True)
assert display.get("depthcue")
with pytest.raises(TypeError):
display.set(depthcueue="hello")
def test_3dgetset():
"""
tests: eyesep, focallength, height, distance, nearclip, farclip,
depthcueue, culling, stereo, projection
"""
modes = display.stereomodes()
assert len(modes) == 11
display.set(eyesep=2.0,
focallength=10.0,
height=200,
distance=20,
stereo=modes[0])
assert display.get("eyesep") == 2.0
assert display.get("focallength") == 10.0
assert display.get("height") == 200
assert display.get("distance") == 20
assert display.get("stereo") == modes[0]
# Default values from DisplayDevice.C
assert display.get("nearclip") == 0.5
assert display.get("farclip") == 10.0
display.set(nearclip=10.0, farclip=30.0)
assert display.get("nearclip") == 10.0
assert display.get("farclip") == 30.0
with pytest.raises(ValueError):
display.set(nearclip=10.0, farclip=5.0)
display.set(culling=False)
assert not display.get("culling")
with pytest.raises(TypeError):
display.set(culling=5.0)
assert "QuadBuffered" in modes
display.set(stereo="QuadBuffered")
assert display.get("stereo") == "QuadBuffered"
display.set(projection=display.PROJ_PERSP)
assert display.get("projection") == display.PROJ_PERSP
with pytest.raises(ValueError):
display.set(projection="invalid")
display.set(depthcue=True)
assert display.get("depthcue")
with pytest.raises(TypeError):
display.set(depthcueue="hello")