コード例 #1
0
ファイル: utilities.py プロジェクト: mdavezac/LaDa
def test_periodic():
  """ Test periodic images. """
  from pylada.crystal.cppwrappers import are_periodic_images
  from numpy import array, dot
  from numpy.linalg import inv
  from random import randint, random
  cell = array([ [0, 0.5, 0.5], [0.5, 0, 0.5], [0.5, 0.5, 0] ])
  invcell = inv(cell)
  for i in range(10):
    vec = dot(cell, array([random(), random(), random()]))
    for j in range(10):
      vec2 = vec + dot(cell, array([randint(-10, 11), randint(-10, 11), randint(-10, 11)]))
      assert are_periodic_images(vec, vec2, invcell)
      vec3 = vec2 + dot(cell, array([random()+0.0001, random(), random()]))
      assert not are_periodic_images(vec, vec3, invcell)
コード例 #2
0
ファイル: utilities.py プロジェクト: spiningup/pylada-light
def test_periodic():
    """ Test periodic images. """
    from pylada.crystal.cppwrappers import are_periodic_images
    from numpy import array, dot
    from numpy.linalg import inv
    from random import randint, random
    cell = array([[0, 0.5, 0.5], [0.5, 0, 0.5], [0.5, 0.5, 0]])
    invcell = inv(cell)
    for i in range(10):
        vec = dot(cell, array([random(), random(), random()]))
        for j in range(10):
            vec2 = vec + dot(
                cell,
                array([randint(-10, 11),
                       randint(-10, 11),
                       randint(-10, 11)]))
            assert are_periodic_images(vec, vec2, invcell)
            vec3 = vec2 + dot(
                cell, array([random() + 0.0001,
                             random(), random()]))
            assert not are_periodic_images(vec, vec3, invcell)
コード例 #3
0
ファイル: utilities.py プロジェクト: mdavezac/LaDa
def test_zero_centered():
  """ Test that vector is folded to fractional coordinates >= -0.5 and < 0.5. """
  from pylada.crystal.cppwrappers import are_periodic_images, zero_centered
  from numpy import array, dot, all
  from numpy.linalg import inv
  from random import uniform
  cell = array([ [0, 0.5, 0.5], [0.5, 0, 0.5], [0.5, 0.5, 0] ])
  invcell = inv(cell)
  for i in range(100):
    vec = dot(cell, array([uniform(-10, 10), uniform(-10, 10), uniform(-10, 10)]))
    vec2 = zero_centered(vec, cell, invcell)
    assert are_periodic_images(vec, vec2, invcell)
    assert all(dot(invcell, vec2) >= -0.5) and all(dot(invcell, vec2) < 0.5)
コード例 #4
0
ファイル: utilities.py プロジェクト: spiningup/pylada-light
def test_into_cell():
    """ Test that vector is folded to fractional coordinates >= 0 and < 1. """
    from pylada.crystal.cppwrappers import are_periodic_images, into_cell
    from numpy import array, dot, all
    from numpy.linalg import inv
    from random import uniform
    cell = array([[0, 0.5, 0.5], [0.5, 0, 0.5], [0.5, 0.5, 0]])
    invcell = inv(cell)
    for i in range(100):
        vec = dot(
            cell, array([uniform(-10, 10),
                         uniform(-10, 10),
                         uniform(-10, 10)]))
        vec2 = into_cell(vec, cell, invcell)
        assert are_periodic_images(vec, vec2, invcell)
        assert all(dot(invcell, vec2) >= 0e0) and all(dot(invcell, vec2) < 1e0)
コード例 #5
0
ファイル: utilities.py プロジェクト: mdavezac/LaDa
def test_into_voronoi():
  """ Test that vector is folded into first Brillouin zone. """
  from pylada.crystal.cppwrappers import are_periodic_images, into_voronoi
  from numpy import array, dot
  from numpy.linalg import inv
  from random import uniform
  cell = array([ [0, 0.5, 0.5], [0.5, 0, 0.5], [0.5, 0.5, 0] ])
  invcell = inv(cell)
  for i in range(100):
    vec = dot(cell, array([uniform(-10, 10), uniform(-10, 10), uniform(-10, 10)]))
    vec2 = into_voronoi(vec, cell, invcell)
    assert are_periodic_images(vec, vec2, invcell)
    n = dot(vec2, vec2)
    assert dot(vec, vec) >= n
    for j in range(-3, 4):
      for k in range(-3, 4):
        for l in range(-3, 4):
          o = vec2 + dot(cell, [j, k, l])
          assert dot(o, o) >= n
コード例 #6
0
ファイル: utilities.py プロジェクト: spiningup/pylada-light
def test_into_voronoi():
    """ Test that vector is folded into first Brillouin zone. """
    from pylada.crystal.cppwrappers import are_periodic_images, into_voronoi
    from numpy import array, dot
    from numpy.linalg import inv
    from random import uniform
    cell = array([[0, 0.5, 0.5], [0.5, 0, 0.5], [0.5, 0.5, 0]])
    invcell = inv(cell)
    for i in range(100):
        vec = dot(
            cell, array([uniform(-10, 10),
                         uniform(-10, 10),
                         uniform(-10, 10)]))
        vec2 = into_voronoi(vec, cell, invcell)
        assert are_periodic_images(vec, vec2, invcell)
        n = dot(vec2, vec2)
        assert dot(vec, vec) >= n
        for j in range(-3, 4):
            for k in range(-3, 4):
                for l in range(-3, 4):
                    o = vec2 + dot(cell, [j, k, l])
                    assert dot(o, o) >= n