def test_9(self):
from phenum.grouptheory import get_sym_group
from phenum.phonons import add_arrows
col = [[-1, 1], [-1, 3], [-1, 2], [1, 4], [1, 2], [-1, 4], [-1, 3],
[-1, 1]]
agroup = _read_output("add_arrow_group.in.9")
dim = 6
out = [[[-1, 1], [-1, 3], [-1, 2], [0, 4], [0, 2], [-1, 4], [-1, 3],
[-1, 1]],
[[-1, 1], [-1, 3], [-1, 2], [1, 4], [0, 2], [-1, 4], [-1, 3],
[-1, 1]],
[[-1, 1], [-1, 3], [-1, 2], [2, 4], [0, 2], [-1, 4], [-1, 3],
[-1, 1]],
[[-1, 1], [-1, 3], [-1, 2], [5, 4], [0, 2], [-1, 4], [-1, 3],
[-1, 1]],
[[-1, 1], [-1, 3], [-1, 2], [0, 4], [1, 2], [-1, 4], [-1, 3],
[-1, 1]],
[[-1, 1], [-1, 3], [-1, 2], [1, 4], [1, 2], [-1, 4], [-1, 3],
[-1, 1]],
[[-1, 1], [-1, 3], [-1, 2], [2, 4], [1, 2], [-1, 4], [-1, 3],
[-1, 1]],
[[-1, 1], [-1, 3], [-1, 2], [4, 4], [1, 2], [-1, 4], [-1, 3],
[-1, 1]],
[[-1, 1], [-1, 3], [-1, 2], [0, 4], [2, 2], [-1, 4], [-1, 3],
[-1, 1]],
[[-1, 1], [-1, 3], [-1, 2], [1, 4], [2, 2], [-1, 4], [-1, 3],
[-1, 1]],
[[-1, 1], [-1, 3], [-1, 2], [2, 4], [2, 2], [-1, 4], [-1, 3],
[-1, 1]],
[[-1, 1], [-1, 3], [-1, 2], [3, 4], [2, 2], [-1, 4], [-1, 3],
[-1, 1]]]
self.assertEqual(
add_arrows(col, agroup, dim, agroup[0:len(col)], supers=True), out)
def test_10(self):
from phenum.grouptheory import get_sym_group
from phenum.phonons import add_arrows
col = [[-1, 1], [-1, 2], [-1, 3], [1, 2], [-1, 3], [-1, 4], [-1, 1], [1, 4]]
agroup = _read_output("add_arrow_group.in.10")
dim = 6
out = [[[-1, 1], [-1, 2], [-1, 3], [0, 2], [-1, 3], [-1, 4], [-1, 1], [0, 4]], [[-1, 1], [-1, 2], [-1, 3], [1, 2], [-1, 3], [-1, 4], [-1, 1], [0, 4]], [[-1, 1], [-1, 2], [-1, 3], [2, 2], [-1, 3], [-1, 4], [-1, 1], [0, 4]], [[-1, 1], [-1, 2], [-1, 3], [5, 2], [-1, 3], [-1, 4], [-1, 1], [0, 4]], [[-1, 1], [-1, 2], [-1, 3], [0, 2], [-1, 3], [-1, 4], [-1, 1], [1, 4]], [[-1, 1], [-1, 2], [-1, 3], [1, 2], [-1, 3], [-1, 4], [-1, 1], [1, 4]], [[-1, 1], [-1, 2], [-1, 3], [2, 2], [-1, 3], [-1, 4], [-1, 1], [1, 4]], [[-1, 1], [-1, 2], [-1, 3], [4, 2], [-1, 3], [-1, 4], [-1, 1], [1, 4]], [[-1, 1], [-1, 2], [-1, 3], [5, 2], [-1, 3], [-1, 4], [-1, 1], [1, 4]], [[-1, 1], [-1, 2], [-1, 3], [0, 2], [-1, 3], [-1, 4], [-1, 1], [2, 4]], [[-1, 1], [-1, 2], [-1, 3], [1, 2], [-1, 3], [-1, 4], [-1, 1], [2, 4]], [[-1, 1], [-1, 2], [-1, 3], [2, 2], [-1, 3], [-1, 4], [-1, 1], [2, 4]], [[-1, 1], [-1, 2], [-1, 3], [3, 2], [-1, 3], [-1, 4], [-1, 1], [2, 4]], [[-1, 1], [-1, 2], [-1, 3], [5, 2], [-1, 3], [-1, 4], [-1, 1], [2, 4]], [[-1, 1], [-1, 2], [-1, 3], [0, 2], [-1, 3], [-1, 4], [-1, 1], [5, 4]], [[-1, 1], [-1, 2], [-1, 3], [1, 2], [-1, 3], [-1, 4], [-1, 1], [5, 4]], [[-1, 1], [-1, 2], [-1, 3], [2, 2], [-1, 3], [-1, 4], [-1, 1], [5, 4]], [[-1, 1], [-1, 2], [-1, 3], [5, 2], [-1, 3], [-1, 4], [-1, 1], [5, 4]]]
self.assertEqual(add_arrows(col,agroup,dim,agroup[0:len(col)],supers=True),out)