def test_get_states_number(self):
    g1 = NetworkGraph(self.s1.structure)
    g1.add_nodes(self.s1.structure.nodes_labels)
    g1.add_edges(self.s1.structure.edges)
    for (node, val) in zip(g1.nodes, g1.nodes_values):
        self.assertEqual(val, g1.get_states_number(node))
def test_get_states_number(self):
    g1 = NetworkGraph(self.s1.structure)
    g1.add_nodes(self.s1.structure.nodes_labels)
    g1.add_edges(self.s1.structure.edges)
    for (node, val) in zip(g1.nodes, g1.nodes_values):
        self.assertEqual(val, g1.get_states_number(node))