Anders and Briegel in Python
Du kannst nicht mehr als 25 Themen auswählen Themen müssen entweder mit einem Buchstaben oder einer Ziffer beginnen. Sie können Bindestriche („-“) enthalten und bis zu 35 Zeichen lang sein.

96 Zeilen
2.5KB

  1. from abp import GraphState
  2. from abp import clifford
  3. from demograph import demograph
  4. import time
  5. def test_graph_basic():
  6. """ Test that we can construct graphs, delete edges, whatever """
  7. g = demograph()
  8. assert g.ngbh[0] == set([1, 2, 3])
  9. g.del_edge(0, 1)
  10. assert g.ngbh[0] == set([2, 3])
  11. assert g.has_edge(1, 2)
  12. assert not g.has_edge(0, 1)
  13. def test_local_complementation():
  14. """ Test that local complementation works as expected """
  15. g = demograph()
  16. g.local_complementation(0)
  17. assert g.has_edge(0, 1)
  18. assert g.has_edge(0, 2)
  19. assert not g.has_edge(1, 2)
  20. assert g.has_edge(3, 2)
  21. assert g.has_edge(3, 1)
  22. # TODO: test VOP conditions
  23. #def test_remove_vop_simple():
  24. #""" Test that removing VOPs really works """
  25. #g = GraphState(xrange(2))
  26. #print g
  27. #g.remove_vop(0, 1)
  28. #print g
  29. #assert g.vops[0] == clifford.by_name["identity"]
  30. #g.remove_vop(1, 1)
  31. #assert g.vops[1] == clifford.by_name["identity"]
  32. #g.remove_vop(2, 1)
  33. #assert g.vops[2] == clifford.by_name["identity"]
  34. #g.remove_vop(0, 1)
  35. #assert g.vops[0] == clifford.by_name["identity"]
  36. def test_remove_vop():
  37. """ Test that removing VOPs really works """
  38. g = demograph()
  39. g.remove_vop(0, 1)
  40. assert g.vops[0] == clifford.by_name["identity"]
  41. g.remove_vop(1, 1)
  42. assert g.vops[1] == clifford.by_name["identity"]
  43. g.remove_vop(2, 1)
  44. assert g.vops[2] == clifford.by_name["identity"]
  45. g.remove_vop(0, 1)
  46. assert g.vops[0] == clifford.by_name["identity"]
  47. def test_edgelist():
  48. """ Test making edgelists """
  49. g = demograph()
  50. el = g.edgelist()
  51. assert (0, 3) in el
  52. assert (0, 2) in el
  53. assert (100, 200) in el
  54. def test_stress(n = int(1e5)):
  55. """ Testing that making a graph of ten thousand qubits takes less than half a second"""
  56. g = GraphState(range(n+1))
  57. t = time.clock()
  58. for i in xrange(n):
  59. g.add_edge(i, i + 1)
  60. assert time.clock() - t < .5
  61. def test_cz():
  62. """ Test CZ gate """
  63. g = GraphState([0, 1])
  64. g.add_node(0)
  65. g.add_node(1)
  66. g.act_local_rotation(0, clifford.by_name["hadamard"])
  67. g.act_local_rotation(1, clifford.by_name["hadamard"])
  68. g.act_local_rotation(1, clifford.by_name["py"])
  69. assert not g.has_edge(0, 1)
  70. g.act_cz(0, 1)
  71. assert g.has_edge(0, 1)
  72. def test_stabilizer():
  73. """ Test that we can generate stabilizers okay """
  74. g = demograph()
  75. stab = g.to_stabilizer()
  76. #TODO: sux
  77. #assert len(stab.split("\n")) == g.order()