def test_disconnected(self): bqm = dimod.BQM.from_ising({}, {'ab': -1, 'ac': -1, 'de': -1}) cc = list(dimod.connected_components(bqm)) self.assertIn( cc, [[set(['a', 'b', 'c']), set(['d', 'e'])], [set(['d', 'e']), set(['a', 'b', 'c'])]])
def test_connected(self): bqm = dimod.BQM.from_ising({}, {'ab': -1, 'ac': -1, 'bd': -1}) cc = list(dimod.connected_components(bqm)) self.assertEqual(cc, [set(bqm.variables)])