def test_cstate_sort1():
state = StateC(ar(10, 10), ar(10))
sort = state.sort()
assert len(state) == len(sort)
c = sort.c[::-1]
sort_descending = sort.sort(False)
assert np.allclose(c, sort_descending.c)
def test_cstate_repr1():
state = StateC(ar(6), 1)
assert len(state) == 1
str(state)
state = StateC(ar(6), 1, parent=geom.graphene())
str(state)
assert len(state) == 1
def test_cstate_sub1():
state = StateC(ar(10, 10), ar(10))
assert len(state) == 10
norm = state.norm()
for i in range(len(state)):
assert len(state.sub(i)) == 1
assert state.sub(i).norm()[0] == norm[i]
assert state[i].norm()[0] == norm[i]
assert state[i].c[0] == state.c[i]
for i, sub in enumerate(state):
assert len(sub) == 1
assert sub.norm()[0] == norm[i]
def test_cstate_outer1():
state = ar(10, 10)
state = StateC(state, ar(10))
out = state.outer()
o = out.copy()
o1 = out.copy()
o.fill(0)
o1.fill(0)
for i, sub in enumerate(state):
o += couter(sub.c[0], sub.state[0, :])
o1 += state.outer(i)
assert np.allclose(out, o)
assert np.allclose(out, o1)
o = state.outer(np.arange(len(state)))
assert np.allclose(out, o)
def test_cstate_sub():
state = StateC(ar(10, 10), ar(10))
assert len(state) == 10
norm = state.norm()
norm2 = state.norm2()
for i in range(len(state)):
assert len(state.sub(i)) == 1
assert state.sub(i).norm()[0] == norm[i]
assert state[i].norm()[0] == norm[i]
assert state[i].norm2()[0] == norm2[i]
assert state[i].c[0] == state.c[i]
for i, sub in enumerate(state):
assert len(sub) == 1
assert sub.norm()[0] == norm[i]
sub = state.sub(np.array([False, True, False, True]))
state.sub([1, 3], inplace=True)
assert np.allclose(sub.c, state.c)
assert np.allclose(sub.state, state.state)
def test_cstate_creation1():
state = StateC(ar(6), 1)
assert len(state) == 1
state = StateC(ar(6, 6), ar(6))
assert len(state) == 6
assert np.allclose(state.c, ar(6))
str(state)
state2 = state.copy()
assert np.allclose(state2.c, state.c)
assert np.allclose(state2.state, state.state)
state2 = state.asState()
assert np.allclose(state2.state, state.state)
state2 = state.asCoefficient()
assert np.allclose(state2.c, state.c)
def test_cstate_norm1():
state = StateC(ar(10, 10), ar(10)).normalize()
assert len(state) == 10
assert np.allclose(state.norm(), 1)
assert np.allclose(state.norm2(), 1)
