This repository has been archived on 2023-03-25. You can view files and clone it, but cannot push or open issues or pull requests.
mightyscape-1.1-deprecated/extensions/fablabchemnitz/networkx/classes/tests/test_special.py
2020-08-30 12:36:33 +02:00

167 lines
4.9 KiB
Python

#!/usr/bin/env python
from collections import OrderedDict
import networkx as nx
from .test_graph import TestGraph
from .test_digraph import TestDiGraph
from .test_multigraph import TestMultiGraph
from .test_multidigraph import TestMultiDiGraph
def test_factories():
class mydict1(dict):
pass
class mydict2(dict):
pass
class mydict3(dict):
pass
class mydict4(dict):
pass
class mydict5(dict):
pass
for Graph in (nx.Graph, nx.DiGraph, nx.MultiGraph, nx.MultiDiGraph):
# print("testing class: ", Graph.__name__)
class MyGraph(Graph):
node_dict_factory = mydict1
adjlist_outer_dict_factory = mydict2
adjlist_inner_dict_factory = mydict3
edge_key_dict_factory = mydict4
edge_attr_dict_factory = mydict5
G = MyGraph()
assert isinstance(G._node, mydict1)
assert isinstance(G._adj, mydict2)
G.add_node(1)
assert isinstance(G._adj[1], mydict3)
if G.is_directed():
assert isinstance(G._pred, mydict2)
assert isinstance(G._succ, mydict2)
assert isinstance(G._pred[1], mydict3)
G.add_edge(1, 2)
if G.is_multigraph():
assert isinstance(G._adj[1][2], mydict4)
assert isinstance(G._adj[1][2][0], mydict5)
else:
assert isinstance(G._adj[1][2], mydict5)
class SpecialGraphTester(TestGraph):
def setUp(self):
TestGraph.setUp(self)
self.Graph = nx.Graph
class OrderedGraphTester(TestGraph):
def setUp(self):
TestGraph.setUp(self)
class MyGraph(nx.Graph):
node_dict_factory = OrderedDict
adjlist_outer_dict_factory = OrderedDict
adjlist_inner_dict_factory = OrderedDict
edge_attr_dict_factory = OrderedDict
self.Graph = MyGraph
class ThinGraphTester(TestGraph):
def setUp(self):
all_edge_dict = {'weight': 1}
class MyGraph(nx.Graph):
def edge_attr_dict_factory(self): return all_edge_dict
self.Graph = MyGraph
# build dict-of-dict-of-dict K3
ed1, ed2, ed3 = (all_edge_dict, all_edge_dict, all_edge_dict)
self.k3adj = {0: {1: ed1, 2: ed2},
1: {0: ed1, 2: ed3},
2: {0: ed2, 1: ed3}}
self.k3edges = [(0, 1), (0, 2), (1, 2)]
self.k3nodes = [0, 1, 2]
self.K3 = self.Graph()
self.K3._adj = self.k3adj
self.K3._node = {}
self.K3._node[0] = {}
self.K3._node[1] = {}
self.K3._node[2] = {}
class SpecialDiGraphTester(TestDiGraph):
def setUp(self):
TestDiGraph.setUp(self)
self.Graph = nx.DiGraph
class OrderedDiGraphTester(TestDiGraph):
def setUp(self):
TestGraph.setUp(self)
class MyGraph(nx.DiGraph):
node_dict_factory = OrderedDict
adjlist_outer_dict_factory = OrderedDict
adjlist_inner_dict_factory = OrderedDict
edge_attr_dict_factory = OrderedDict
self.Graph = MyGraph
class ThinDiGraphTester(TestDiGraph):
def setUp(self):
all_edge_dict = {'weight': 1}
class MyGraph(nx.DiGraph):
def edge_attr_dict_factory(self): return all_edge_dict
self.Graph = MyGraph
# build dict-of-dict-of-dict K3
ed1, ed2, ed3 = (all_edge_dict, all_edge_dict, all_edge_dict)
self.k3adj = {0: {1: ed1, 2: ed2},
1: {0: ed1, 2: ed3},
2: {0: ed2, 1: ed3}}
self.k3edges = [(0, 1), (0, 2), (1, 2)]
self.k3nodes = [0, 1, 2]
self.K3 = self.Graph()
self.K3.adj = self.k3adj
self.K3._node = {}
self.K3._node[0] = {}
self.K3._node[1] = {}
self.K3._node[2] = {}
class SpecialMultiGraphTester(TestMultiGraph):
def setUp(self):
TestMultiGraph.setUp(self)
self.Graph = nx.MultiGraph
class OrderedMultiGraphTester(TestMultiGraph):
def setUp(self):
TestMultiGraph.setUp(self)
class MyGraph(nx.MultiGraph):
node_dict_factory = OrderedDict
adjlist_outer_dict_factory = OrderedDict
adjlist_inner_dict_factory = OrderedDict
edge_key_dict_factory = OrderedDict
edge_attr_dict_factory = OrderedDict
self.Graph = MyGraph
class SpecialMultiDiGraphTester(TestMultiDiGraph):
def setUp(self):
TestMultiDiGraph.setUp(self)
self.Graph = nx.MultiDiGraph
class OrderedMultiDiGraphTester(TestMultiDiGraph):
def setUp(self):
TestMultiDiGraph.setUp(self)
class MyGraph(nx.MultiDiGraph):
node_dict_factory = OrderedDict
adjlist_outer_dict_factory = OrderedDict
adjlist_inner_dict_factory = OrderedDict
edge_key_dict_factory = OrderedDict
edge_attr_dict_factory = OrderedDict
self.Graph = MyGraph