def test_empty(self):
sampler = PolyScaleComposite(RangeLimitedSampler())
samples = sampler.sample_hubo({})
self.assertEqual(len(samples.variables), 0)
sampler = PolyScaleComposite(RangeLimitedSampler())
samples = sampler.sample_hubo({'a': 4}, scalar=.25)
self.assertEqual(samples.first.energy, 4)
def test_fail_scale(self):
sampler = PolyScaleComposite(RangeLimitedSampler())
with self.assertRaises(RuntimeError):
sampler.sample_hubo({'a': 4}, scalar=1)