Anders and Briegel in Python
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

37 line
846B

  1. import json
  2. import networkx as nx
  3. from abp import GraphState, fancy
  4. from abp import clifford
  5. from abp.util import xyz
  6. from mock import simple_graph
  7. def test_json_basic():
  8. """ Test that we can export to JSON """
  9. g = simple_graph()
  10. js = g.to_json()
  11. assert "adj" in js
  12. assert "node" in js
  13. def test_tuple_keys():
  14. """ Test that we can use tuple-ish keys """
  15. g = fancy.GraphState()
  16. g.add_node("string")
  17. g.add_node((1, 2, 3))
  18. g.add_edge((1, 2, 3), "string")
  19. json.dumps(g.to_json(True))
  20. def networkx_test():
  21. """ Test that fancy graph states really behave like networkx graphs """
  22. g = fancy.GraphState()
  23. g.add_node(0, position = xyz(10, 0, 0))
  24. g.add_node(1, position = xyz(1, 0, 0))
  25. g.act_hadamard(0)
  26. g.act_hadamard(1)
  27. g.act_cz(0, 1)
  28. g.copy()
  29. # TODO: more tests here!