def test_not_implemented(): class Test: pass t = Test() with pytest.raises(NotImplementedError): sisl.plot(t)
def test_orbital_radial(): r = np.linspace(0, 10, 1000) f = np.exp(- r) o = sisl.SphericalOrbital(2, (r, f)) sisl.plot(o) fig = plt.figure() sisl.plot(o, axes=fig.gca()) plt.close('all')
def test_geometry_2d(): g = sisl.geom.graphene() sisl.plot(g, axis=[0, 1]) sisl.plot(g, axis=[0, 2]) sisl.plot(g, axis=[1, 2]) ax = plt.subplot(111) sisl.plot(g, axis=[1, 2], axes=ax) plt.close('all')
def test_supercell_2d(): g = sisl.geom.graphene() sisl.plot(g.sc, axis=[0, 1]) sisl.plot(g.sc, axis=[0, 2]) sisl.plot(g.sc, axis=[1, 2]) ax = plt.subplot(111) sisl.plot(g.sc, axis=[1, 2], axes=ax) plt.close('all')
def test_geometry_2d_atom_indices(): g = sisl.geom.graphene() sisl.plot(g, axis=[0, 1]) sisl.plot(g, axis=[0, 2]) sisl.plot(g, axis=[1, 2]) ax = plt.subplot(111) sisl.plot(g, axis=[1, 2], axes=ax, atom_indices=True) plt.close('all')
def test_orbital_harmonics(): r = np.linspace(0, 10, 1000) f = np.exp(-r) o = sisl.SphericalOrbital(2, (r, f)) sisl.plot(o, harmonics=True) plt.close('all')
def test_geometry_3d_atom_indices(): g = sisl.geom.graphene() sisl.plot(g, atom_indices=True) plt.close('all')
def test_geometry_3d(): g = sisl.geom.graphene() sisl.plot(g) plt.close('all')
def test_supercell_3d(): g = sisl.geom.graphene() sisl.plot(g.sc) plt.close('all')
def test_not_implemented(): class Test(object): pass t = Test() sisl.plot(t)