fix some warnings in tests

Author: refraction-ray

pytest.ini

@@ -0,0 +1,4 @@
+[pytest]
+filterwarnings =
+    ignore::DeprecationWarning
+    ignore:Explicitly requested dtype*:UserWarning 

tensorcircuit/abstractcircuit.py

@@ -745,7 +745,7 @@ def vis_tex(self, **kws: Any) -> str:
         """
         if (not self.is_mps) and (self.inputs is not None):
             init = ["" for _ in range(self._nqubits)]
-            init[self._nqubits // 2] = "\psi"
+            init[self._nqubits // 2] = r"\psi"
             okws = {"init": init}
         else:
             okws = {"init": None}  # type: ignore

tensorcircuit/applications/graphdata.py

@@ -290,7 +290,7 @@ def regular_graph_generator(d: int, n: int, weights: bool = False) -> Iterator[G
 
 
 def _maxcut(g: Graph, values: Sequence[int]) -> float:
-    """
+    r"""
     To get the max-cut for the graph g by given values $$\pm 1$$ on each vertex as a list.
 
     :param g: The graph

tensorcircuit/applications/layers.py

@@ -81,7 +81,7 @@ def f(
 
 
 def generate_gate_layer(gate: str) -> None:
-    """
+    r"""
     $$e^{-i\theta \sigma}$$
 
     :param gate:
@@ -108,7 +108,7 @@ def f(
 
 
 def generate_any_gate_layer(gate: str) -> None:
-    """
+    r"""
     $$e^{-i\theta_i \sigma}$$
 
     :param gate:
@@ -392,7 +392,7 @@ def cirqcnotgate(
         return circuit
 
     def generate_cirq_gate_layer(gate: str) -> None:
-        """
+        r"""
         $$e^{-i\theta \sigma}$$
 
         :param gate:
@@ -422,7 +422,7 @@ def f(
         setattr(thismodule, "cirq" + gate + "layer", f)
 
     def generate_cirq_any_gate_layer(gate: str) -> None:
-        """
+        r"""
         $$e^{-i\theta \sigma}$$
 
         :param gate:

tensorcircuit/asciiart.py

@@ -38,7 +38,7 @@ def __str__(self) -> str:
 (*)`(*)
 """
 
-__bigbike__ = """
+__bigbike__ = r"""
                                           $"   *.      
               d$$$$$$$P"                  $    J
                   ^$.                     4r  "
@@ -58,7 +58,7 @@ def __str__(self) -> str:
        "*==*""                             ^"*==*""   Gilo94'
 """
 
-__moon__ = """
+__moon__ = r"""
          ___---___                    
       .--         --.      
     ./   ()      .-. \.
@@ -77,7 +77,7 @@ def __str__(self) -> str:
 """
 
 
-__cat__ = """
+__cat__ = r"""
  ,_     _
  |\ _,-~/
  / _  _ |    ,--.
@@ -90,7 +90,7 @@ def __str__(self) -> str:
  ((_/`(____,-'
 """
 
-__moonlanding__ = """
+__moonlanding__ = r"""
                      _  _     ____________.--.
                   |\|_|//_.-"" .'    \   /|  |
                   |.-"-"-.|   /       \_/ |  |
@@ -124,7 +124,7 @@ def __str__(self) -> str:
 
 """
 
-__moonshot__ = """
+__moonshot__ = r"""
                          .-.
                         ( (
                          `-'

tensorcircuit/backends/jax_backend.py

@@ -395,7 +395,7 @@ def is_tensor(self, a: Any) -> bool:
         return False
 
     def solve(self, A: Tensor, b: Tensor, assume_a: str = "gen") -> Tensor:  # type: ignore
-        return jsp.linalg.solve(A, b, assume_a)
+        return jsp.linalg.solve(A, b, assume_a=assume_a)
 
     def searchsorted(self, a: Tensor, v: Tensor, side: str = "left") -> Tensor:
         if not self.is_tensor(a):
@@ -408,10 +408,10 @@ def tree_map(self, f: Callable[..., Any], *pytrees: Any) -> Any:
         return libjax.tree_map(f, *pytrees)
 
     def tree_flatten(self, pytree: Any) -> Tuple[Any, Any]:
-        return libjax.tree_flatten(pytree)  # type: ignore
+        return libjax.tree_util.tree_flatten(pytree)  # type: ignore
 
     def tree_unflatten(self, treedef: Any, leaves: Any) -> Any:
-        return libjax.tree_unflatten(treedef, leaves)
+        return libjax.tree_util.tree_unflatten(treedef, leaves)
 
     def from_dlpack(self, a: Any) -> Tensor:
         import jax.dlpack

tensorcircuit/backends/numpy_backend.py

@@ -229,7 +229,7 @@ def left_shift(self, x: Tensor, y: Tensor) -> Tensor:
     def solve(self, A: Tensor, b: Tensor, assume_a: str = "gen") -> Tensor:  # type: ignore
         # gen, sym, her, pos
         # https://stackoverflow.com/questions/44672029/difference-between-numpy-linalg-solve-and-numpy-linalg-lu-solve/44710451
-        return solve(A, b, assume_a)
+        return solve(A, b, assume_a=assume_a)
 
     def searchsorted(self, a: Tensor, v: Tensor, side: str = "left") -> Tensor:
         return np.searchsorted(a, v, side=side)  # type: ignore

tensorcircuit/quantum.py

@@ -711,7 +711,7 @@ def eval(
         return list(nodes)[0].tensor
 
     def eval_matrix(self, final_edge_order: Optional[Sequence[Edge]] = None) -> Tensor:
-        """
+        r"""
         Contracts the tensor network in place and returns the final tensor
         in two dimentional matrix.
         The default ordering for the axes of the final tensor is:
@@ -2176,7 +2176,7 @@ def spin_by_basis(n: int, m: int, elements: Tuple[int, int] = (1, -1)) -> Tensor
 
 
 def correlation_from_samples(index: Sequence[int], results: Tensor, n: int) -> Tensor:
-    """
+    r"""
     Compute :math:`\prod_{i\in \\text{index}} s_i (s=\pm 1)`,
     Results is in the format of "sample_int" or "sample_bin"
 
@@ -2200,7 +2200,7 @@ def correlation_from_samples(index: Sequence[int], results: Tensor, n: int) -> T
 
 
 def correlation_from_counts(index: Sequence[int], results: Tensor) -> Tensor:
-    """
+    r"""
     Compute :math:`\prod_{i\in \\text{index}} s_i`,
     where the probability for each bitstring is given as a vector ``results``.
     Results is in the format of "count_vector"

tensorcircuit/templates/measurements.py

@@ -218,7 +218,7 @@ def heisenberg_measurements(
     hy: float = 0.0,
     reuse: bool = True,
 ) -> Tensor:
-    """
+    r"""
     Evaluate Heisenberg energy expectation, whose Hamiltonian is defined on the lattice graph ``g`` as follows:
     (e are edges in graph ``g`` where e1 and e2 are two nodes for edge e and v are nodes in graph ``g``)
 

tensorcircuit/utils.py

@@ -136,7 +136,7 @@ def wrapper(*args: Any, **kws: Any) -> Any:
             if isinstance(vs, str):
                 vs = []
             for v in vs:
-                if kws.get(v, "qazxswedcvfr198") != "qazxswedcvfr198":
+                if v in kws:
                     # in case it is None by design!
                     kws[k] = kws[v]
                     del kws[v]

tensorcircuit/vis.py

@@ -102,7 +102,7 @@ def qir2tex(
             p = max(flag[min(idx) : max(idx) + 1]) + 1
             for i in range(min(idx), max(idx) + 1):
                 tex_string_table[i] += [r"\qw "] * (p - flag[i] - 1)
-                tex_string_table[i] += [r"\ghost{" + x["name"][7:] + "}\qw "]
+                tex_string_table[i] += [r"\ghost{" + x["name"][7:] + r"}\qw "]
                 flag[i] = p
         else:
             ctrl_number, gate_name = gate_name_trans(x["name"])