Fix #215: Handle non-contiguous numbering in rustworkx backend

graphix/graphsim/rxgraphstate.py

@@ -81,7 +81,7 @@ def neighbors(self, node) -> Iterator:
  See :meth:`BaseGraphState.neighbors`.
  """
  nidx = self.nodes.get_node_index(node)
- return iter(self._graph.neighbors(nidx))
+ return iter(self.nodes.idx_to_num[idx] for idx in self._graph.neighbors(nidx))
 
  def subgraph(self, nodes: list) -> rx.PyGraph:
  """Return a subgraph of the graph.

graphix/graphsim/rxgraphviews.py

@@ -31,6 +31,7 @@ def __init__(
  self.nodes = set(node_nums)
  self.num_to_data = {nnum: node_datas[nidx] for nidx, nnum in zip(node_indices, node_nums)}
  self.num_to_idx = {nnum: nidx for nidx, nnum in zip(node_indices, node_nums)}
+ self.idx_to_num = {nidx: nnum for nidx, nnum in zip(node_indices, node_nums)}
 
  def __contains__(self, nnum: int) -> bool:
  """Return `True` if the node `nnum` belongs to the list, `False` otherwise."""
@@ -64,6 +65,7 @@ def add_node(self, nnum: int, ndata: dict, nidx: int) -> None:
  self.nodes.add(nnum)
  self.num_to_data[nnum] = ndata
  self.num_to_idx[nnum] = nidx
+ self.idx_to_num[nidx] = nnum
 
  def add_nodes_from(self, node_nums: list[int], node_datas: list[dict], node_indices: list[int]) -> None:
  """Add nodes to the list."""
@@ -80,7 +82,8 @@ def remove_node(self, nnum: int) -> None:
  raise ValueError(f"Node {nnum} does not exist")
  self.nodes.remove(nnum)
  del self.num_to_data[nnum]
- del self.num_to_idx[nnum]
+ idx = self.num_to_idx.pop(nnum)
+ del self.idx_to_num[idx]
 
  def remove_nodes_from(self, node_nums: list[int]) -> None:
  """Remove nodes from the list."""

tests/test_pattern.py

@@ -78,6 +78,18 @@ def test_minimize_space(self, fx_rng: Generator) -> None:
  state_mbqc = pattern.simulate_pattern(rng=fx_rng)
  assert np.abs(np.dot(state_mbqc.flatten().conjugate(), state.flatten())) == pytest.approx(1)
 
+ @pytest.mark.parametrize("use_rustworkx", [False, True])
+ def test_pauli_non_contiguous(self, use_rustworkx) -> None:
+ pattern = Pattern(input_nodes=[0])
+ pattern.extend(
+ [
+ N(node=2, state=PlanarState(plane=Plane.XY, angle=0.0)),
+ E(nodes=(0, 2)),
+ M(node=0, plane=Plane.XY, angle=0.0, s_domain=set(), t_domain=set()),
+ ]
+ )
+ pattern.perform_pauli_measurements(use_rustworkx=use_rustworkx)
+
  @pytest.mark.parametrize("jumps", range(1, 11))
  def test_minimize_space_with_gflow(self, fx_bg: PCG64, jumps: int, use_rustworkx: bool = True) -> None:
  rng = Generator(fx_bg.jumped(jumps))