Line data    Source code

       1              : # SPDX-FileCopyrightText: 2026 PairInteraction Developers
       2              : # SPDX-License-Identifier: LGPL-3.0-or-later
       3              : 
       4            1 : from __future__ import annotations
       5              : 
       6            1 : from typing import TYPE_CHECKING
       7              : 
       8              : if TYPE_CHECKING:
       9              :     from .utils import PairinteractionModule
      10              : 
      11              : 
      12            1 : def test_state_pair_label(pi_module: PairinteractionModule) -> None:
      13              :     """Test the label of a pair eigenstate."""
      14            1 :     basis_atom = pi_module.BasisAtom("Rb", n=(60, 60), l=(0, 1))
      15            1 :     system_atom = pi_module.SystemAtom(basis_atom).set_electric_field([0, 0, 5], unit="V/cm").diagonalize()
      16            1 :     basis_pair = pi_module.BasisPair([system_atom, system_atom], m=(1, 1))
      17            1 :     system_pair = pi_module.SystemPair(basis_pair).diagonalize()
      18              : 
      19            1 :     state = system_pair.get_eigenbasis().get_state(0)
      20            1 :     expected_label = (
      21              :         "0.77 |Rb:60,S_1/2,1/2; Rb:60,S_1/2,1/2⟩ - 0.34 |Rb:60,P_3/2,1/2; Rb:60,S_1/2,1/2⟩ "
      22              :         "- 0.34 |Rb:60,S_1/2,1/2; Rb:60,P_3/2,1/2⟩ + ..."
      23              :     )
      24              : 
      25            1 :     assert state.get_label() == expected_label