test_random_clustered.py 979 B

12345678910111213141516171819202122232425262728293031323334
  1. import pytest
  2. import networkx
  3. class TestRandomClusteredGraph:
  4. def test_valid(self):
  5. node = [1, 1, 1, 2, 1, 2, 0, 0]
  6. tri = [0, 0, 0, 0, 0, 1, 1, 1]
  7. joint_degree_sequence = zip(node, tri)
  8. G = networkx.random_clustered_graph(joint_degree_sequence)
  9. assert G.number_of_nodes() == 8
  10. assert G.number_of_edges() == 7
  11. def test_valid2(self):
  12. G = networkx.random_clustered_graph(
  13. [(1, 2), (2, 1), (1, 1), (1, 1), (1, 1), (2, 0)]
  14. )
  15. assert G.number_of_nodes() == 6
  16. assert G.number_of_edges() == 10
  17. def test_invalid1(self):
  18. pytest.raises(
  19. (TypeError, networkx.NetworkXError),
  20. networkx.random_clustered_graph,
  21. [[1, 1], [2, 1], [0, 1]],
  22. )
  23. def test_invalid2(self):
  24. pytest.raises(
  25. (TypeError, networkx.NetworkXError),
  26. networkx.random_clustered_graph,
  27. [[1, 1], [1, 2], [0, 1]],
  28. )