def test_poschl_teller_eigen_energy_float_lambda(self, level, lam,
                                                  expected_energy):
     # NOTE(leeley): I calculated some eigen energy for float lambda by the
     # expression in Phys. Rev. B 95, 115115
     self.assertAlmostEqual(
         single_electron.poschl_teller_eigen_energy(level, lam),
         expected_energy)
 def test_poschl_teller_eigen_energy_int_lambda(
     self, level, lam, expected_energy):
   # For integer value of lambda, the eigen energy is
   # E = -\mu^2 / 2
   # \mu = \lambda, \lambda - 1, ..., 1
   # https://en.wikipedia.org/wiki/P%C3%B6schl%E2%80%93Teller_potential
   self.assertAlmostEqual(
       single_electron.poschl_teller_eigen_energy(level, lam), expected_energy)