Python get_point_moments Examples

Example #1

def test_rotate_cartesian_moments_random_general():
    for i in xrange(10):
        coordinates = np.random.normal(0, 1, (10, 3))
        rmat = get_random_rotation()
        m0 = get_point_moments(coordinates)
        m1 = get_point_moments(coordinates, rmat)
        m2 = rotate_cartesian_moments_all(rmat, m0)
        assert abs(m1 - m2).max() < 1e-10

Example #2

File: test_moments.py Project: QuantumElephant/horton

def test_rotate_cartesian_moments_random_general():
    for i in xrange(10):
        coordinates = np.random.normal(0, 1, (10, 3))
        rmat = get_random_rotation()
        m0 = get_point_moments(coordinates)
        m1 = get_point_moments(coordinates, rmat)
        m2 = rotate_cartesian_moments_all(rmat, m0)
        assert abs(m1 - m2).max() < 1e-10

Example #3

def test_rotate_cartesian_moments_random_mult():
    for i in xrange(10):
        coordinates = np.random.normal(0, 1, (10, 3))
        for mult in 2, 3, 4, 5:
            axis = np.random.normal(0, 1, 3)
            rmat = get_rotation_matrix(axis, np.pi / mult)
            m0 = get_point_moments(coordinates)
            m1 = get_point_moments(coordinates)
            for i in xrange(mult):
                m1 = rotate_cartesian_moments_all(rmat, m1)
                m1 = rotate_cartesian_moments_all(rmat, m1)
            assert abs(m0 - m1).max() < 1e-10

Example #4

File: test_moments.py Project: QuantumElephant/horton

def test_rotate_cartesian_moments_random_mult():
    for i in xrange(10):
        coordinates = np.random.normal(0, 1, (10, 3))
        for mult in 2, 3, 4, 5:
            axis = np.random.normal(0, 1, 3)
            rmat = get_rotation_matrix(axis, np.pi/mult)
            m0 = get_point_moments(coordinates)
            m1 = get_point_moments(coordinates)
            for i in xrange(mult):
                m1 = rotate_cartesian_moments_all(rmat, m1)
                m1 = rotate_cartesian_moments_all(rmat, m1)
            assert abs(m0 - m1).max() < 1e-10