def test_state_align_norm1():
state = ortho_matrix(10)
state1 = State(state)
idx = np.arange(len(state))
np.random.shuffle(idx)
state2 = state1.sub(idx)
# This should swap all back
align2 = state1.align_norm(state2)
assert np.allclose(state1.state, align2.state)
def test_state_align_norm2():
state = ortho_matrix(15)
state1 = State(state)
idx = np.arange(len(state))
np.random.shuffle(idx)
state2 = state1.sub(idx)
# This should swap all back
align2, idx2 = state1.align_norm(state2, ret_index=True)
assert np.allclose(state1.state, align2.state)
assert np.allclose(state1.state, state2.sub(idx2).state)
