def test_default_set(self):
dnx.set_default_sampler(MockSampler())
mock_function(0)
dnx.unset_default_sampler()
self.assertEqual(dnx.get_default_sampler(), None,
"sampler did not unset correctly")
def test_default_sampler(self):
G = nx.complete_graph(5)
dnx.set_default_sampler(dimod.ExactSolver())
self.assertIsNot(dnx.get_default_sampler(), None)
indep_set = dnx.maximum_independent_set(G)
dnx.unset_default_sampler()
self.assertEqual(dnx.get_default_sampler(), None, "sampler did not unset correctly")
def test_default_sampler(self):
G = nx.complete_graph(5)
dnx.set_default_sampler(ExactSolver())
self.assertIsNot(dnx.get_default_sampler(), None)
cover = dnx.min_vertex_cover(G)
dnx.unset_default_sampler()
self.assertEqual(dnx.get_default_sampler(), None,
"sampler did not unset correctly")
def test_default_sampler(self):
G = nx.complete_graph(5)
dnx.set_default_sampler(ExactSolver())
self.assertIsNot(dnx.get_default_sampler(), None)
matching = dnx.algorithms.matching.maximal_matching(G)
matching = dnx.min_maximal_matching(G)
dnx.unset_default_sampler()
self.assertEqual(dnx.get_default_sampler(), None,
"sampler did not unset correctly")
def test_default_sampler(self):
S = nx.Graph()
S.add_edge('Alice', 'Bob', sign=1) # Alice and Bob are friendly
S.add_edge('Alice', 'Eve', sign=-1) # Alice and Eve are hostile
S.add_edge('Bob', 'Eve', sign=-1) # Bob and Eve are hostile
dnx.set_default_sampler(ExactSolver())
self.assertIsNot(dnx.get_default_sampler(), None)
frustrated_edges, colors = dnx.structural_imbalance(S)
dnx.unset_default_sampler()
self.assertEqual(dnx.get_default_sampler(), None, "sampler did not unset correctly")
