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.

test_fancy.py 846B

123456789101112131415161718192021222324252627282930313233343536
  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!