def test_sample_two_components(self):
bqm = BinaryQuadraticModel({
0: 0.0,
1: 4.0,
2: -4.0,
3: 0.0
}, {
(0, 1): -4.0,
(2, 3): 4.0
}, 0.0, Vartype.BINARY)
sampler = ConnectedComponentsComposite(ExactSolver())
response = sampler.sample(bqm)
self.assertIsInstance(response, SampleSet)
self.assertEqual(response.first.sample, {0: 0, 1: 0, 2: 1, 3: 0})
self.assertAlmostEqual(response.first.energy,
bqm.energy({
0: 0,
1: 0,
2: 1,
3: 0
}))
def test_sample_passcomponents(self):
bqm = BinaryQuadraticModel({
0: 0.0,
1: 4.0,
2: -4.0,
3: 0.0
}, {
(0, 1): -4.0,
(2, 3): 4.0
}, 0.0, Vartype.BINARY)
with self.assertWarns(DeprecationWarning):
sampler = ConnectedComponentsComposite(ExactSolver())
response = sampler.sample(bqm, components=[{0, 1}, {2, 3}])
self.assertIsInstance(response, SampleSet)
self.assertEqual(response.first.sample, {0: 0, 1: 0, 2: 1, 3: 0})
self.assertAlmostEqual(response.first.energy,
bqm.energy({
0: 0,
1: 0,
2: 1,
3: 0
}))
