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)