def test_sample_qubo_variables(self, mock_client):
mock_client.from_config.side_effect = MockClient
sampler = LeapHybridSampler()
response = sampler.sample_qubo({(0, 0): -1, (1, 1): 1})
rows, cols = response.record.sample.shape
self.assertEqual(cols, 2)
response = sampler.sample_qubo({(0, 0): -1, (1, 1): 1})
rows, cols = response.record.sample.shape
self.assertEqual(cols, 2)
self.assertTrue(np.all(response.record.sample >= 0))
self.assertIs(response.vartype, dimod.BINARY)
def run_annealer_hybrid(theta):
sampler = LeapHybridSampler()
response = sampler.sample_qubo(theta)
response = response.first.sample.values()
return np.atleast_2d(list(response)).T