Anders and Briegel in Python
25개 이상의 토픽을 선택하실 수 없습니다. Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

test_nx.py 1.1KB

7 년 전
7 년 전
12345678910111213141516171819202122232425262728293031323334353637383940414243444546
  1. import json
  2. import networkx as nx
  3. from abp import GraphState, NXGraphState
  4. from abp import clifford
  5. from abp.util import xyz
  6. import pytest
  7. mock = pytest.importorskip("mock")
  8. def test_json_basic():
  9. """ Test that we can export to JSON """
  10. g = mock.simple_graph()
  11. js = g.to_json()
  12. assert "adj" in js
  13. assert "node" in js
  14. def test_tuple_keys():
  15. """ Test that we can use tuple-ish keys """
  16. g = NXGraphState()
  17. g.add_qubit("string")
  18. g.add_qubit((1, 2, 3))
  19. g.add_edge((1, 2, 3), "string")
  20. json.dumps(g.to_json(True))
  21. def networkx_test():
  22. """ Test that NXGraphStates really behave like networkx graphs """
  23. g = NXGraphState()
  24. g.add_qubit(0, position=xyz(10, 0, 0))
  25. g.add_qubit(1, position=xyz(1, 0, 0))
  26. g.act_hadamard(0)
  27. g.act_hadamard(1)
  28. g.act_cz(0, 1)
  29. g.copy()
  30. def test_from_nx():
  31. """ Test that making graphs from networkx objects goes smoothly """
  32. g = nx.random_geometric_graph(100, 2)
  33. psi = NXGraphState(g)
  34. assert psi.node[0]["vop"] == 0
  35. assert len(psi.edges()) > 0
  36. psi.measure(0, "px", detail=True)
  37. psi = NXGraphState(nx.Graph(((0, 1),)))