Coverage for jaxquantum/codes/mode.py: 62%

1""" 

2Generic Bosonic Mode Class 

3""" 

4 

5from typing import Tuple 

6 

7from jaxquantum.codes.base import BosonicQubit 

8import jaxquantum as jqt 

9 

10from jax import config 

11 

12config.update("jax_enable_x64", True) 

13 

14 

15class BosonicMode(BosonicQubit): 

16 """ 

17 FockQubit 

18 """ 

19 

20 def _params_validation(self): 

21 super()._params_validation() 

22 

23 def _get_basis_z(self) -> Tuple[jqt.Qarray, jqt.Qarray]: 

24 """ 

25 Construct basis states |+-x>, |+-y>, |+-z> 

26 """ 

27 N = int(self.params["N"]) 

28 plus_z = jqt.basis(N, 0) 

29 minus_z = jqt.basis(N, 1) 

30 return plus_z, minus_z