def test_state_align_phase():
state = ortho_matrix(10)
state1 = State(state)
state2 = State(-state)
# This should rotate all back
align2 = state2.align_phase(state1)
assert np.allclose(state1.state, align2.state)
align2, idx = state2.align_phase(state1, ret_index=True)
assert not np.allclose(state1.state, state2.state)
state2.align_phase(state1, inplace=True)
assert np.allclose(state1.state, state2.state)
state2 = State(-state)
idx = state2.align_phase(state1, inplace=True, ret_index=True)
assert np.allclose(state1.state, state2.state)
def test_state_align_phase1():
state = ortho_matrix(10)
state1 = State(state)
state2 = State(-state)
# This should rotate all back
align2 = state1.align_phase(state2)
assert np.allclose(state1.state, align2.state)
