123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205 |
- import pytest
- import networkx as nx
- from networkx.algorithms.isomorphism.isomorph import graph_could_be_isomorphic
- is_isomorphic = graph_could_be_isomorphic
- """Generators - Small
- =====================
- Some small graphs
- """
- null = nx.null_graph()
- class TestGeneratorsSmall:
- def test__LCF_graph(self):
- # If n<=0, then return the null_graph
- G = nx.LCF_graph(-10, [1, 2], 100)
- assert is_isomorphic(G, null)
- G = nx.LCF_graph(0, [1, 2], 3)
- assert is_isomorphic(G, null)
- G = nx.LCF_graph(0, [1, 2], 10)
- assert is_isomorphic(G, null)
- # Test that LCF(n,[],0) == cycle_graph(n)
- for a, b, c in [(5, [], 0), (10, [], 0), (5, [], 1), (10, [], 10)]:
- G = nx.LCF_graph(a, b, c)
- assert is_isomorphic(G, nx.cycle_graph(a))
- # Generate the utility graph K_{3,3}
- G = nx.LCF_graph(6, [3, -3], 3)
- utility_graph = nx.complete_bipartite_graph(3, 3)
- assert is_isomorphic(G, utility_graph)
- def test_properties_named_small_graphs(self):
- G = nx.bull_graph()
- assert sorted(G) == list(range(5))
- assert G.number_of_edges() == 5
- assert sorted(d for n, d in G.degree()) == [1, 1, 2, 3, 3]
- assert nx.diameter(G) == 3
- assert nx.radius(G) == 2
- G = nx.chvatal_graph()
- assert sorted(G) == list(range(12))
- assert G.number_of_edges() == 24
- assert [d for n, d in G.degree()] == 12 * [4]
- assert nx.diameter(G) == 2
- assert nx.radius(G) == 2
- G = nx.cubical_graph()
- assert sorted(G) == list(range(8))
- assert G.number_of_edges() == 12
- assert [d for n, d in G.degree()] == 8 * [3]
- assert nx.diameter(G) == 3
- assert nx.radius(G) == 3
- G = nx.desargues_graph()
- assert sorted(G) == list(range(20))
- assert G.number_of_edges() == 30
- assert [d for n, d in G.degree()] == 20 * [3]
- G = nx.diamond_graph()
- assert sorted(G) == list(range(4))
- assert sorted(d for n, d in G.degree()) == [2, 2, 3, 3]
- assert nx.diameter(G) == 2
- assert nx.radius(G) == 1
- G = nx.dodecahedral_graph()
- assert sorted(G) == list(range(20))
- assert G.number_of_edges() == 30
- assert [d for n, d in G.degree()] == 20 * [3]
- assert nx.diameter(G) == 5
- assert nx.radius(G) == 5
- G = nx.frucht_graph()
- assert sorted(G) == list(range(12))
- assert G.number_of_edges() == 18
- assert [d for n, d in G.degree()] == 12 * [3]
- assert nx.diameter(G) == 4
- assert nx.radius(G) == 3
- G = nx.heawood_graph()
- assert sorted(G) == list(range(14))
- assert G.number_of_edges() == 21
- assert [d for n, d in G.degree()] == 14 * [3]
- assert nx.diameter(G) == 3
- assert nx.radius(G) == 3
- G = nx.hoffman_singleton_graph()
- assert sorted(G) == list(range(50))
- assert G.number_of_edges() == 175
- assert [d for n, d in G.degree()] == 50 * [7]
- assert nx.diameter(G) == 2
- assert nx.radius(G) == 2
- G = nx.house_graph()
- assert sorted(G) == list(range(5))
- assert G.number_of_edges() == 6
- assert sorted(d for n, d in G.degree()) == [2, 2, 2, 3, 3]
- assert nx.diameter(G) == 2
- assert nx.radius(G) == 2
- G = nx.house_x_graph()
- assert sorted(G) == list(range(5))
- assert G.number_of_edges() == 8
- assert sorted(d for n, d in G.degree()) == [2, 3, 3, 4, 4]
- assert nx.diameter(G) == 2
- assert nx.radius(G) == 1
- G = nx.icosahedral_graph()
- assert sorted(G) == list(range(12))
- assert G.number_of_edges() == 30
- assert [d for n, d in G.degree()] == [5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5]
- assert nx.diameter(G) == 3
- assert nx.radius(G) == 3
- G = nx.krackhardt_kite_graph()
- assert sorted(G) == list(range(10))
- assert G.number_of_edges() == 18
- assert sorted(d for n, d in G.degree()) == [1, 2, 3, 3, 3, 4, 4, 5, 5, 6]
- G = nx.moebius_kantor_graph()
- assert sorted(G) == list(range(16))
- assert G.number_of_edges() == 24
- assert [d for n, d in G.degree()] == 16 * [3]
- assert nx.diameter(G) == 4
- G = nx.octahedral_graph()
- assert sorted(G) == list(range(6))
- assert G.number_of_edges() == 12
- assert [d for n, d in G.degree()] == 6 * [4]
- assert nx.diameter(G) == 2
- assert nx.radius(G) == 2
- G = nx.pappus_graph()
- assert sorted(G) == list(range(18))
- assert G.number_of_edges() == 27
- assert [d for n, d in G.degree()] == 18 * [3]
- assert nx.diameter(G) == 4
- G = nx.petersen_graph()
- assert sorted(G) == list(range(10))
- assert G.number_of_edges() == 15
- assert [d for n, d in G.degree()] == 10 * [3]
- assert nx.diameter(G) == 2
- assert nx.radius(G) == 2
- G = nx.sedgewick_maze_graph()
- assert sorted(G) == list(range(8))
- assert G.number_of_edges() == 10
- assert sorted(d for n, d in G.degree()) == [1, 2, 2, 2, 3, 3, 3, 4]
- G = nx.tetrahedral_graph()
- assert sorted(G) == list(range(4))
- assert G.number_of_edges() == 6
- assert [d for n, d in G.degree()] == [3, 3, 3, 3]
- assert nx.diameter(G) == 1
- assert nx.radius(G) == 1
- G = nx.truncated_cube_graph()
- assert sorted(G) == list(range(24))
- assert G.number_of_edges() == 36
- assert [d for n, d in G.degree()] == 24 * [3]
- G = nx.truncated_tetrahedron_graph()
- assert sorted(G) == list(range(12))
- assert G.number_of_edges() == 18
- assert [d for n, d in G.degree()] == 12 * [3]
- G = nx.tutte_graph()
- assert sorted(G) == list(range(46))
- assert G.number_of_edges() == 69
- assert [d for n, d in G.degree()] == 46 * [3]
- # Test create_using with directed or multigraphs on small graphs
- pytest.raises(nx.NetworkXError, nx.tutte_graph, create_using=nx.DiGraph)
- MG = nx.tutte_graph(create_using=nx.MultiGraph)
- assert sorted(MG.edges()) == sorted(G.edges())
- @pytest.mark.parametrize(
- "fn",
- (
- nx.bull_graph,
- nx.chvatal_graph,
- nx.cubical_graph,
- nx.diamond_graph,
- nx.house_graph,
- nx.house_x_graph,
- nx.icosahedral_graph,
- nx.krackhardt_kite_graph,
- nx.octahedral_graph,
- nx.petersen_graph,
- nx.truncated_cube_graph,
- nx.tutte_graph,
- ),
- )
- @pytest.mark.parametrize(
- "create_using", (nx.DiGraph, nx.MultiDiGraph, nx.DiGraph([(0, 1)]))
- )
- def tests_raises_with_directed_create_using(fn, create_using):
- with pytest.raises(nx.NetworkXError, match="Directed Graph not supported"):
- fn(create_using=create_using)
|