def test_sparse_orbital_transform_nonortho_so():
M = SparseOrbitalBZSpin(geom.graphene(), spin='so', orthogonal=False)
a = np.arange(M.spin.size + 1) + 0.3
M.construct(([0.1, 1.44], [a, a + 0.1]))
M.finalize()
Mcsr = [M.tocsr(i) for i in range(M.shape[2])]
Mt = M.transform(spin='unpolarized')
assert np.abs(0.5 * Mcsr[0] + 0.5 * Mcsr[1] - Mt.tocsr(0)).sum() == 0
assert np.abs(Mcsr[-1] - Mt.tocsr(-1)).sum() == 0
Mt = M.transform(spin='polarized')
assert np.abs(Mcsr[0] - Mt.tocsr(0)).sum() == 0
assert np.abs(Mcsr[1] - Mt.tocsr(1)).sum() == 0
assert np.abs(Mcsr[-1] - Mt.tocsr(-1)).sum() == 0
Mt = M.transform(spin='non-colinear')
assert np.abs(Mcsr[0] - Mt.tocsr(0)).sum() == 0
assert np.abs(Mcsr[1] - Mt.tocsr(1)).sum() == 0
assert np.abs(Mcsr[2] - Mt.tocsr(2)).sum() == 0
assert np.abs(Mcsr[3] - Mt.tocsr(3)).sum() == 0
assert np.abs(Mcsr[-1] - Mt.tocsr(-1)).sum() == 0
Mt = M.transform(spin='so')
assert np.abs(Mcsr[0] - Mt.tocsr(0)).sum() == 0
assert np.abs(Mcsr[1] - Mt.tocsr(1)).sum() == 0
assert np.abs(Mcsr[2] - Mt.tocsr(2)).sum() == 0
assert np.abs(Mcsr[3] - Mt.tocsr(3)).sum() == 0
assert np.abs(Mcsr[-1] - Mt.tocsr(-1)).sum() == 0
def test_sparse_orbital_transform_fail():
M = SparseOrbitalBZSpin(geom.graphene(),
spin='polarized',
orthogonal=False,
dtype=np.int32)
M.construct(([0.1, 1.44], [(1, 2, 3), (4, 5, 6)]))
M.finalize()
with pytest.raises(ValueError):
M.transform(np.zeros([2, 2]), spin='unpolarized')
def test_sparse_orbital_transform_combinations():
M = SparseOrbitalBZSpin(geom.graphene(),
spin='polarized',
orthogonal=False,
dtype=np.int32)
M.construct(([0.1, 1.44], [(3, 2, 1), (2, 1, 0)]))
M.finalize()
Mcsr = [M.tocsr(i) for i in range(M.shape[2])]
Mt = M.transform(spin='non-colinear', dtype=np.float64,
orthogonal=True).transform(spin='polarized',
orthogonal=False)
assert M.dim == Mt.dim
assert np.abs(Mcsr[0] - Mt.tocsr(0)).sum() == 0
assert np.abs(Mcsr[1] - Mt.tocsr(1)).sum() == 0
assert np.abs(Mcsr[-1] - Mt.tocsr(-1)).sum() == 0
Mt = M.transform(dtype=np.float128,
orthogonal=True).transform(spin='so',
dtype=np.float64,
orthogonal=False)
assert np.abs(Mcsr[0] - Mt.tocsr(0)).sum() == 0
assert np.abs(Mcsr[1] - Mt.tocsr(1)).sum() == 0
assert np.abs(Mt.tocsr(2)).sum() == 0
assert np.abs(Mcsr[-1] - Mt.tocsr(-1)).sum() == 0
Mt = M.transform(spin='polarized',
orthogonal=True).transform(spin='so',
dtype=np.float64,
orthogonal=False)
assert np.abs(Mcsr[0] - Mt.tocsr(0)).sum() == 0
assert np.abs(Mcsr[1] - Mt.tocsr(1)).sum() == 0
assert np.abs(Mt.tocsr(2)).sum() == 0
assert np.abs(Mcsr[-1] - Mt.tocsr(-1)).sum() == 0
Mt = M.transform(spin='unpolarized', dtype=np.float32,
orthogonal=True).transform(dtype=np.complex128,
orthogonal=False)
assert np.abs(0.5 * Mcsr[0] + 0.5 * Mcsr[1] - Mt.tocsr(0)).sum() == 0
def test_sparse_orbital_transform_basis():
M = SparseOrbitalBZSpin(geom.graphene(),
spin='polarized',
orthogonal=False)
M.construct(([0.1, 1.44], [(3., 2., 1.), (0.3, 0.2, 0.)]))
M.finalize()
Mcsr = [M.tocsr(i) for i in range(M.shape[2])]
Mt = M.transform(orthogonal=True).transform(orthogonal=False)
assert M.dim == Mt.dim
assert np.abs(Mcsr[0] - Mt.tocsr(0)).sum() == 0
assert np.abs(Mcsr[1] - Mt.tocsr(1)).sum() == 0
assert np.abs(Mcsr[-1] - Mt.tocsr(-1)).sum() == 0
def test_sparse_orbital_transform_matrix():
M = SparseOrbitalBZSpin(geom.graphene(),
spin='polarized',
orthogonal=False,
dtype=np.int32)
M.construct(([0.1, 1.44], [(1, 2, 3), (4, 5, 6)]))
M.finalize()
Mcsr = [M.tocsr(i) for i in range(M.shape[2])]
Mt = M.transform(spin='unpolarized',
matrix=np.ones((1, 3)),
orthogonal=True)
assert Mt.dim == 1
assert np.abs(Mcsr[0] + Mcsr[1] + Mcsr[2] - Mt.tocsr(0)).sum() == 0
Mt = M.transform(spin='polarized', matrix=np.ones((2, 3)), orthogonal=True)
assert Mt.dim == 2
assert np.abs(Mcsr[0] + Mcsr[1] + Mcsr[2] - Mt.tocsr(1)).sum() == 0
Mt = M.transform(matrix=np.ones((3, 3)), dtype=np.float64)
assert Mt.dim == 3
assert np.abs(Mcsr[0] + Mcsr[1] + Mcsr[2] - Mt.tocsr(2)).sum() == 0
Mt = M.transform(spin='non-colinear',
matrix=np.ones((4, 3)),
orthogonal=True,
dtype=np.float64)
assert Mt.dim == 4
assert np.abs(Mcsr[0] + Mcsr[1] + Mcsr[2] - Mt.tocsr(3)).sum() == 0
Mt = M.transform(spin='non-colinear',
matrix=np.ones((5, 3)),
dtype=np.float64)
assert Mt.dim == 5
assert np.abs(Mcsr[0] + Mcsr[1] + Mcsr[2] - Mt.tocsr(4)).sum() == 0
Mt = M.transform(spin='so',
matrix=np.ones((8, 3)),
orthogonal=True,
dtype=np.float64)
assert Mt.dim == 8
assert np.abs(Mcsr[0] + Mcsr[1] + Mcsr[2] - Mt.tocsr(7)).sum() == 0
Mt = M.transform(spin='so', matrix=np.ones((9, 3)), dtype=np.float64)
assert Mt.dim == 9
assert np.abs(Mcsr[0] + Mcsr[1] + Mcsr[2] - Mt.tocsr(8)).sum() == 0
