def test_state_sub1():
state = ar(10, 10)
state = State(state)
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]
for i, sub in enumerate(state):
assert len(sub) == 1
assert sub.norm()[0] == norm[i]
for i, sub in enumerate(state.iter(True)):
assert (sub**2).sum()**.5 == norm[i]
def test_state_sub1():
state = ar(10, 10)
state = State(state)
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]
for i, sub in enumerate(state):
assert len(sub) == 1
assert sub.norm()[0] == norm[i]
for i, sub in enumerate(state.iter(True)):
assert (sub ** 2).sum() == norm2[i]
assert np.allclose(state.sub([False, True, False, True]).state, state.sub([1, 3]).state)
