class SQTensorTest(PymatgenTest):
def setUp(self):
self.rand_sqtensor = SQTensor(np.random.randn(3, 3))
self.symm_sqtensor = SQTensor([[0.1, 0.3, 0.4], [0.3, 0.5, 0.2],
[0.4, 0.2, 0.6]])
self.non_invertible = SQTensor([[0.1, 0, 0], [0.2, 0, 0], [0, 0, 0]])
self.non_symm = SQTensor([[0.1, 0.2, 0.3], [0.4, 0.5, 0.6],
[0.2, 0.5, 0.5]])
self.low_val = SQTensor([[1e-6, 1 + 1e-5,
1e-6], [1 + 1e-6, 1e-6, 1e-6],
[1e-7, 1e-7, 1 + 1e-5]])
self.low_val_2 = SQTensor([[1e-6, -1 - 1e-6, 1e-6],
[1 + 1e-7, 1e-6, 1e-6],
[1e-7, 1e-7, 1 + 1e-6]])
a = 3.14 * 42.5 / 180
self.rotation = SQTensor([[math.cos(a), 0, math.sin(a)], [0, 1, 0],
[-math.sin(a), 0,
math.cos(a)]])
def test_new(self):
non_sq_matrix = [[0.1, 0.2, 0.1], [0.1, 0.2, 0.3], [0.1, 0.2, 0.3],
[0.1, 0.1, 0.1]]
bad_matrix = [[0.1, 0.2], [0.2, 0.3, 0.4], [0.2, 0.3, 0.5]]
self.assertRaises(ValueError, SQTensor, non_sq_matrix)
self.assertRaises(ValueError, SQTensor, bad_matrix)
def test_properties(self):
# transpose
self.assertArrayEqual(
self.non_symm.trans,
SQTensor([[0.1, 0.4, 0.2], [0.2, 0.5, 0.5], [0.3, 0.6, 0.5]]))
self.assertArrayEqual(self.rand_sqtensor.trans,
np.transpose(self.rand_sqtensor))
self.assertArrayEqual(self.symm_sqtensor, self.symm_sqtensor.trans)
# inverse
self.assertArrayEqual(self.non_symm.inv, np.linalg.inv(self.non_symm))
with self.assertRaises(ValueError):
self.non_invertible.inv
# determinant
self.assertEqual(self.rand_sqtensor.det,
np.linalg.det(self.rand_sqtensor))
self.assertEqual(self.non_invertible.det, 0.0)
self.assertEqual(self.non_symm.det, 0.009)
# symmetrized
self.assertArrayEqual(
self.rand_sqtensor.symmetrized,
0.5 * (self.rand_sqtensor + self.rand_sqtensor.trans))
self.assertArrayEqual(self.symm_sqtensor,
self.symm_sqtensor.symmetrized)
self.assertArrayAlmostEqual(
self.non_symm.symmetrized,
SQTensor([[0.1, 0.3, 0.25], [0.3, 0.5, 0.55], [0.25, 0.55, 0.5]]))
# invariants
i1 = np.trace(self.rand_sqtensor)
i2 = self.rand_sqtensor[0, 0] * self.rand_sqtensor[1, 1] + \
self.rand_sqtensor[1, 1] * self.rand_sqtensor[2, 2] + \
self.rand_sqtensor[2, 2] * self.rand_sqtensor[0, 0] - \
self.rand_sqtensor[0, 1] * self.rand_sqtensor[1, 0] - \
self.rand_sqtensor[0, 2] * self.rand_sqtensor[2, 0] - \
self.rand_sqtensor[2, 1] * self.rand_sqtensor[1, 2]
i3 = np.linalg.det(self.rand_sqtensor)
self.assertArrayAlmostEqual([i1, i2, i3],
self.rand_sqtensor.principal_invariants)
def test_symmetry(self):
self.assertTrue(self.symm_sqtensor.is_symmetric())
self.assertFalse(self.non_symm.is_symmetric())
self.assertTrue(self.low_val.is_symmetric())
self.assertFalse(self.low_val.is_symmetric(tol=1e-8))
def test_rotation(self):
self.assertTrue(self.rotation.is_rotation())
self.assertFalse(self.symm_sqtensor.is_rotation())
self.assertTrue(self.low_val_2.is_rotation())
self.assertFalse(self.low_val_2.is_rotation(tol=1e-8))
def test_rotate(self):
self.assertArrayAlmostEqual(self.non_symm.rotate(self.rotation),
SQTensor([[0.531, 0.485, 0.271],
[0.700, 0.5, 0.172],
[0.171, 0.233, 0.068]]),
decimal=3)
self.assertRaises(ValueError, self.non_symm.rotate, self.symm_sqtensor)
def test_get_scaled(self):
self.assertArrayEqual(self.non_symm.get_scaled(10.),
SQTensor([[1, 2, 3], [4, 5, 6], [2, 5, 5]]))
def test_polar_decomposition(self):
u, p = self.rand_sqtensor.polar_decomposition()
self.assertArrayAlmostEqual(np.dot(u, p), self.rand_sqtensor)
self.assertArrayAlmostEqual(np.eye(3),
np.dot(u, np.conjugate(np.transpose(u))))
def test_zeroed(self):
self.assertArrayEqual(
self.low_val.zeroed(),
SQTensor([[0, 1 + 1e-5, 0], [1 + 1e-6, 0, 0], [0, 0, 1 + 1e-5]]))
self.assertArrayEqual(
self.low_val.zeroed(tol=1e-6),
SQTensor([[1e-6, 1 + 1e-5, 1e-6], [1 + 1e-6, 1e-6, 1e-6],
[0, 0, 1 + 1e-5]]))
self.assertArrayEqual(
SQTensor([[1e-6, -30, 1], [1e-7, 1, 0], [1e-8, 0, 1]]).zeroed(),
SQTensor([[0, -30, 1], [0, 1, 0], [0, 0, 1]]))
class SQTensorTest(PymatgenTest):
def setUp(self):
self.rand_sqtensor = SQTensor(np.random.randn(3, 3))
self.symm_sqtensor = SQTensor([[0.1, 0.3, 0.4],
[0.3, 0.5, 0.2],
[0.4, 0.2, 0.6]])
self.non_invertible = SQTensor([[0.1, 0, 0],
[0.2, 0, 0],
[0, 0, 0]])
self.non_symm = SQTensor([[0.1, 0.2, 0.3],
[0.4, 0.5, 0.6],
[0.2, 0.5, 0.5]])
self.low_val = SQTensor([[1e-6, 1 + 1e-5, 1e-6],
[1 + 1e-6, 1e-6, 1e-6],
[1e-7, 1e-7, 1 + 1e-5]])
self.low_val_2 = SQTensor([[1e-6, -1 - 1e-6, 1e-6],
[1 + 1e-7, 1e-6, 1e-6],
[1e-7, 1e-7, 1 + 1e-6]])
a = 3.14 * 42.5 / 180
self.rotation = SQTensor([[math.cos(a), 0, math.sin(a)],
[0, 1, 0],
[-math.sin(a), 0, math.cos(a)]])
def test_new(self):
non_sq_matrix = [[0.1, 0.2, 0.1],
[0.1, 0.2, 0.3],
[0.1, 0.2, 0.3],
[0.1, 0.1, 0.1]]
bad_matrix = [[0.1, 0.2],
[0.2, 0.3, 0.4],
[0.2, 0.3, 0.5]]
self.assertRaises(ValueError, SQTensor, non_sq_matrix)
self.assertRaises(ValueError, SQTensor, bad_matrix)
def test_properties(self):
# transpose
self.assertArrayEqual(self.non_symm.trans, SQTensor([[0.1, 0.4, 0.2],
[0.2, 0.5, 0.5],
[0.3, 0.6, 0.5]]))
self.assertArrayEqual(self.rand_sqtensor.trans,
np.transpose(self.rand_sqtensor))
self.assertArrayEqual(self.symm_sqtensor,
self.symm_sqtensor.trans)
# inverse
self.assertArrayEqual(self.non_symm.inv,
np.linalg.inv(self.non_symm))
with self.assertRaises(ValueError):
self.non_invertible.inv
# determinant
self.assertEqual(self.rand_sqtensor.det,
np.linalg.det(self.rand_sqtensor))
self.assertEqual(self.non_invertible.det,
0.0)
self.assertEqual(self.non_symm.det, 0.009)
# symmetrized
self.assertArrayEqual(self.rand_sqtensor.symmetrized,
0.5 * (self.rand_sqtensor + self.rand_sqtensor.trans))
self.assertArrayEqual(self.symm_sqtensor,
self.symm_sqtensor.symmetrized)
self.assertArrayAlmostEqual(self.non_symm.symmetrized,
SQTensor([[0.1, 0.3, 0.25],
[0.3, 0.5, 0.55],
[0.25, 0.55, 0.5]]))
# invariants
i1 = np.trace(self.rand_sqtensor)
i2 = self.rand_sqtensor[0, 0] * self.rand_sqtensor[1, 1] + \
self.rand_sqtensor[1, 1] * self.rand_sqtensor[2, 2] + \
self.rand_sqtensor[2, 2] * self.rand_sqtensor[0, 0] - \
self.rand_sqtensor[0, 1] * self.rand_sqtensor[1, 0] - \
self.rand_sqtensor[0, 2] * self.rand_sqtensor[2, 0] - \
self.rand_sqtensor[2, 1] * self.rand_sqtensor[1, 2]
i3 = np.linalg.det(self.rand_sqtensor)
self.assertArrayAlmostEqual([i1, i2, i3],
self.rand_sqtensor.principal_invariants)
def test_symmetry(self):
self.assertTrue(self.symm_sqtensor.is_symmetric())
self.assertFalse(self.non_symm.is_symmetric())
self.assertTrue(self.low_val.is_symmetric())
self.assertFalse(self.low_val.is_symmetric(tol=1e-8))
def test_rotation(self):
self.assertTrue(self.rotation.is_rotation())
self.assertFalse(self.symm_sqtensor.is_rotation())
self.assertTrue(self.low_val_2.is_rotation())
self.assertFalse(self.low_val_2.is_rotation(tol=1e-8))
def test_rotate(self):
self.assertArrayAlmostEqual(self.non_symm.rotate(self.rotation),
SQTensor([[0.531, 0.485, 0.271],
[0.700, 0.5, 0.172],
[0.171, 0.233, 0.068]]),
decimal=3)
self.assertRaises(ValueError, self.non_symm.rotate, self.symm_sqtensor)
def test_get_scaled(self):
self.assertArrayEqual(self.non_symm.get_scaled(10.),
SQTensor([[1, 2, 3], [4, 5, 6], [2, 5, 5]]))
def test_polar_decomposition(self):
u, p = self.rand_sqtensor.polar_decomposition()
self.assertArrayAlmostEqual(np.dot(u, p), self.rand_sqtensor)
self.assertArrayAlmostEqual(np.eye(3),
np.dot(u, np.conjugate(np.transpose(u))))
def test_zeroed(self):
self.assertArrayEqual(self.low_val.zeroed(),
SQTensor([[0, 1 + 1e-5, 0],
[1 + 1e-6, 0, 0],
[0, 0, 1 + 1e-5]]))
self.assertArrayEqual(self.low_val.zeroed(tol=1e-6),
SQTensor([[1e-6, 1 + 1e-5, 1e-6],
[1 + 1e-6, 1e-6, 1e-6],
[0, 0, 1 + 1e-5]]))
