From d3e93ca55026c176aee32c106ddc23800bec49e7 Mon Sep 17 00:00:00 2001 From: Pete Shadbolt Date: Thu, 19 May 2016 20:35:19 +0100 Subject: [PATCH] Fix small things --- abp/graphstate.py | 7 +++---- tests/test_nxgraphstate.py | 4 ++-- 2 files changed, 5 insertions(+), 6 deletions(-) diff --git a/abp/graphstate.py b/abp/graphstate.py index 6bf223c..f088b99 100644 --- a/abp/graphstate.py +++ b/abp/graphstate.py @@ -17,13 +17,12 @@ class GraphState(NXGraph): self.adj, self.node = {}, {} self.add_nodes(nodes) - def add_node(self, v, position = None, meta = {}): + def add_node(self, v, **kwargs): """ Add a node """ assert not v in self.node self.adj[v] = {} - self.node[v] = {"vop": clifford.by_name["hadamard"], - "position": position} - self.node[v].update(meta) + self.node[v] = {"vop": clifford.by_name["hadamard"]} + self.node[v].update(kwargs) def add_nodes(self, nodes): """ Add a buncha nodes """ diff --git a/tests/test_nxgraphstate.py b/tests/test_nxgraphstate.py index 3ade04c..3e74667 100644 --- a/tests/test_nxgraphstate.py +++ b/tests/test_nxgraphstate.py @@ -4,8 +4,8 @@ import networkx as nx def simple_test(): g = GraphState() - g.add_node(0, xyz(0, 0, 0)) - g.add_node(1, xyz(0, 0, 0)) + g.add_node(0, position = xyz(10, 0, 0)) + g.add_node(1, position = xyz(1, 0, 0)) g.act_hadamard(0) g.act_hadamard(1) g.act_cz(0, 1)