mirror of
https://github.com/fjosw/pyerrors.git
synced 2025-03-17 07:40:25 +01:00
63 lines
1.8 KiB
Python
63 lines
1.8 KiB
Python
import numpy as np
|
|
import pyerrors as pe
|
|
import pytest
|
|
|
|
np.random.seed(0)
|
|
|
|
|
|
def test_gamma_matrices():
|
|
for matrix in pe.dirac.gamma:
|
|
assert np.allclose(matrix @ matrix, np.identity(4))
|
|
assert np.allclose(matrix, matrix.T.conj())
|
|
assert np.allclose(pe.dirac.gamma5, pe.dirac.gamma[0] @ pe.dirac.gamma[1] @ pe.dirac.gamma[2] @ pe.dirac.gamma[3])
|
|
|
|
|
|
def test_grid_dirac():
|
|
for gamma in ['Identity',
|
|
'Gamma5',
|
|
'GammaX',
|
|
'GammaY',
|
|
'GammaZ',
|
|
'GammaT',
|
|
'GammaXGamma5',
|
|
'GammaYGamma5',
|
|
'GammaZGamma5',
|
|
'GammaTGamma5',
|
|
'SigmaXT',
|
|
'SigmaXY',
|
|
'SigmaXZ',
|
|
'SigmaYT',
|
|
'SigmaYZ',
|
|
'SigmaZT']:
|
|
pe.dirac.Grid_gamma(gamma)
|
|
with pytest.raises(Exception):
|
|
pe.dirac.Grid_gamma('Not a gamma matrix')
|
|
|
|
|
|
def test_epsilon_tensor():
|
|
check = {(1, 2, 3) : 1.0,
|
|
(3, 1, 2) : 1.0,
|
|
(2, 3, 1) : 1.0,
|
|
(1, 1, 1) : 0.0,
|
|
(3, 2, 1) : -1.0,
|
|
(1, 3, 2) : -1.0,
|
|
(1, 1, 3) : 0.0}
|
|
for key, value in check.items():
|
|
assert pe.dirac.epsilon_tensor(*key) == value
|
|
with pytest.raises(Exception):
|
|
pe.dirac.epsilon_tensor(0, 1, 3)
|
|
|
|
|
|
def test_epsilon_tensor_rank4():
|
|
check = {(1, 4, 3, 2) : -1.0,
|
|
(1, 2, 3, 4) : 1.0,
|
|
(2, 1, 3, 4) : -1.0,
|
|
(4, 3, 2, 1) : 1.0,
|
|
(3, 2, 4, 3) : 0.0,
|
|
(0, 1, 2, 3) : 1.0,
|
|
(1, 1, 1, 1) : 0.0,
|
|
(1, 2, 3, 1) : 0.0}
|
|
for key, value in check.items():
|
|
assert pe.dirac.epsilon_tensor_rank4(*key) == value
|
|
with pytest.raises(Exception):
|
|
pe.dirac.epsilon_tensor_rank4(0, 1, 3, 4)
|