def testStrongStableSet(self):
g = nx.Graph()
g.add_node(0)
# check a triangle
result = strong_stable_set(make_clique(3))
self.assertEqual(result.nodes(), g.nodes())
# check a diamond
result = strong_stable_set(make_diamond())
self.assertEqual(result.nodes(), g.nodes())
# C5
result = strong_stable_set(make_cycle(5))
self.assertEqual(result, None)
def testStableSet(self):
expect = [[0], [1], [2, 3]]
for i in stable_set(make_diamond()):
self.assertEqual(i in expect, True)
