test_dominating.py 1.2 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546
  1. import pytest
  2. import networkx as nx
  3. def test_dominating_set():
  4. G = nx.gnp_random_graph(100, 0.1)
  5. D = nx.dominating_set(G)
  6. assert nx.is_dominating_set(G, D)
  7. D = nx.dominating_set(G, start_with=0)
  8. assert nx.is_dominating_set(G, D)
  9. def test_complete():
  10. """In complete graphs each node is a dominating set.
  11. Thus the dominating set has to be of cardinality 1.
  12. """
  13. K4 = nx.complete_graph(4)
  14. assert len(nx.dominating_set(K4)) == 1
  15. K5 = nx.complete_graph(5)
  16. assert len(nx.dominating_set(K5)) == 1
  17. def test_raise_dominating_set():
  18. with pytest.raises(nx.NetworkXError):
  19. G = nx.path_graph(4)
  20. D = nx.dominating_set(G, start_with=10)
  21. def test_is_dominating_set():
  22. G = nx.path_graph(4)
  23. d = {1, 3}
  24. assert nx.is_dominating_set(G, d)
  25. d = {0, 2}
  26. assert nx.is_dominating_set(G, d)
  27. d = {1}
  28. assert not nx.is_dominating_set(G, d)
  29. def test_wikipedia_is_dominating_set():
  30. """Example from https://en.wikipedia.org/wiki/Dominating_set"""
  31. G = nx.cycle_graph(4)
  32. G.add_edges_from([(0, 4), (1, 4), (2, 5)])
  33. assert nx.is_dominating_set(G, {4, 3, 5})
  34. assert nx.is_dominating_set(G, {0, 2})
  35. assert nx.is_dominating_set(G, {1, 2})