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)