def test_bad_problem():
params = dict(VALID_PARAMS)
params.update(num_vars=4,
active_vars=(0, 1, 2, 3),
active_var_pairs=((0, 1), (1, 2), (2, 3), (3, 4)),
var_order=(0, 1, 2, 3))
with pytest.raises(TypeError):
sapilocal.orang_sample('ising', '2 + 2 = ?', params)
with pytest.raises(RuntimeError):
sapilocal.orang_sample('ising', {(0, 2): -1}, params)
def test_qubo_no_hist():
# relying on particular RNG sequence, yikes!
params = {
'num_vars': 4,
'active_vars': range(4),
'active_var_pairs': [[0, 1], [0, 2], [0, 3], [1, 2], [1, 3], [2, 3]],
'var_order': range(4),
'num_reads': 7,
'max_answers': 7,
'answer_histogram': False,
'beta': 0.25,
'random_seed': 10101
}
problem = {
(0, 0): -1,
(1, 1): -1,
(2, 2): -1,
(3, 3): -1,
(0, 1): 0.25,
(0, 2): 0.25,
(0, 3): 0.25,
(1, 2): 0.5,
(1, 3): 0.5,
(2, 3): -1
}
result = sapilocal.orang_sample('qubo', problem, params)
assert result == {
'energies': [-3.00, -1.75, -1.00, -3.00, -1.00, -1.75, -1.75],
'solutions': [[0, 0, 1, 1], [1, 0, 1, 0], [1, 0, 0, 0], [0, 0, 1, 1],
[1, 0, 0, 0], [1, 0, 0, 1], [1, 1, 0, 0]]
}
def test_rng_change():
# params set so that each run takes more than 1ms
# enough variables+runs to ensure distinct outputs on different runs
params = _params(num_vars=8,
active_vars=range(8),
active_var_pairs=[],
var_order=range(8),
num_reads=5000,
max_answers=5000,
answer_histogram=False,
random_seed=None)
problem = dict(((i, i), 0.0) for i in range(8))
r1 = sapilocal.orang_sample('ising', problem, params)
# sometimes for tests run on Windows, r1 == r2, so here add time.sleep(0.01) to get different random seed
time.sleep(0.01)
r2 = sapilocal.orang_sample('ising', problem, params)
assert r1 != r2
def solve(self, problem_type, problem, params):
params2 = dict(self._default_params)
if 'answer_mode' in params:
params2['answer_histogram'] = _is_histmode(params['answer_mode'])
if 'beta' in params:
params2['beta'] = params['beta']
if 'num_reads' in params:
params2['num_reads'] = params['num_reads']
params2['max_answers'] = params.get('max_answers')
if params2['max_answers'] is None:
params2['max_answers'] = params2['num_reads']
params2['random_seed'] = params.get('random_seed')
params2.update(self._config)
return sapilocal.orang_sample(problem_type, problem, params2)
def test_ising_no_hist():
# relying on particular RNG sequence, yikes!
params = {
'num_vars': 5,
'active_vars': range(5),
'active_var_pairs': ((0, 2), (1, 2), (3, 2), (4, 2)),
'var_order': (4, 3, 0, 1, 2),
'num_reads': 6,
'max_answers': 6,
'answer_histogram': False,
'beta': 0.75,
'random_seed': 999
}
problem = {(0, 2): -0.5, (1, 2): -0.5, (2, 4): -0.5}
result = sapilocal.orang_sample('ising', problem, params)
assert result == {
'energies': [-0.5, -1.5, -0.5, 0.5, -0.5, -0.5],
'solutions': [[+1, -1, +1, 3, +1], [-1, -1, -1, 3, -1],
[-1, -1, -1, 3, +1], [+1, -1, -1, 3, +1],
[-1, +1, +1, 3, +1], [-1, -1, -1, 3, +1]]
}
def test_ising_hist():
# relying on particular RNG sequence, yikes!
params = {
'num_vars': 8,
'active_vars': [0, 2, 4, 6],
'active_var_pairs': [(0, 2), (2, 4), (4, 6)],
'var_order': (4, 2, 6, 0),
'num_reads': 10000,
'max_answers': 3,
'answer_histogram': True,
'beta': 1.5,
'random_seed': 11235
}
problem = {(0, 0): 0.25, (0, 2): -1, (2, 4): 1, (4, 6): -0.5}
result = sapilocal.orang_sample('ising', problem, params)
assert result == {
'energies': [-2.75, -2.25, -1.75],
'solutions': [[-1, 3, -1, 3, +1, 3, +1,
3], [+1, 3, +1, 3, -1, 3, -1, 3],
[-1, 3, -1, 3, +1, 3, -1, 3]],
'num_occurrences': [5137, 2354, 1097]
}
def test_qubo_hist():
# relying on particular RNG sequence, yikes!
params = {
'num_vars': 10,
'active_vars': range(9),
'active_var_pairs': [(i, i + 1) for i in range(8)],
'var_order': list(reversed(range(9))),
'num_reads': 8888,
'max_answers': 4,
'answer_histogram': True,
'beta': 0.25,
'random_seed': 4444
}
problem = {
(0, 0): -10,
(0, 1): 3,
(1, 1): 2,
(1, 2): 6,
(2, 2): -4,
(2, 3): -1,
(3, 4): -1,
(4, 5): -1,
(5, 6): -1,
(6, 7): -1,
(7, 8): -1
}
result = sapilocal.orang_sample('qubo', problem, params)
assert result == {
'energies': [-20, -19, -18, -18],
'solutions': [[1, 0, 1, 1, 1, 1, 1, 1, 1, 3],
[1, 0, 1, 1, 1, 1, 1, 1, 0, 3],
[1, 0, 1, 0, 1, 1, 1, 1, 1, 3],
[1, 0, 1, 1, 0, 1, 1, 1, 1, 3]],
'num_occurrences': [228, 190, 156, 100]
}
def test_valid():
sapilocal.orang_sample('ising', {}, VALID_PARAMS)
sapilocal.orang_sample('qubo', {}, VALID_PARAMS)
def test_bad_params():
with pytest.raises(KeyError):
sapilocal.orang_sample('ising', {}, _missing('num_vars'))
with pytest.raises(TypeError):
sapilocal.orang_sample('ising', {}, _params(num_vars='lots'))
with pytest.raises(RuntimeError):
sapilocal.orang_sample('ising', {}, _params(num_vars=-1))
with pytest.raises(KeyError):
sapilocal.orang_sample('ising', {}, _missing('active_vars'))
with pytest.raises(TypeError):
sapilocal.orang_sample('ising', {}, _params(active_vars='all'))
with pytest.raises(RuntimeError):
sapilocal.orang_sample('ising', {}, _params(active_vars=[0]))
with pytest.raises(KeyError):
sapilocal.orang_sample('ising', {}, _missing('active_var_pairs'))
with pytest.raises(TypeError):
sapilocal.orang_sample('ising', {}, _params(active_var_pairs=6))
with pytest.raises(RuntimeError):
sapilocal.orang_sample('ising', {}, _params(active_var_pairs=[(1, 2)]))
with pytest.raises(KeyError):
sapilocal.orang_sample('ising', {}, _missing('var_order'))
with pytest.raises(TypeError):
sapilocal.orang_sample('ising', {}, _params(var_order='all'))
with pytest.raises(RuntimeError):
sapilocal.orang_sample('ising', {}, _params(var_order=[0]))
with pytest.raises(KeyError):
sapilocal.orang_sample('ising', {}, _missing('num_reads'))
with pytest.raises(TypeError):
sapilocal.orang_sample('ising', {}, _params(num_reads=[1, 2, 3]))
with pytest.raises(RuntimeError):
sapilocal.orang_sample('ising', {}, _params(num_reads=-2))
with pytest.raises(KeyError):
sapilocal.orang_sample('ising', {}, _missing('max_answers'))
with pytest.raises(TypeError):
sapilocal.orang_sample('ising', {}, _params(max_answers=[1, 2, 3]))
with pytest.raises(RuntimeError):
sapilocal.orang_sample('ising', {}, _params(max_answers=-2))
with pytest.raises(KeyError):
sapilocal.orang_sample('ising', {}, _missing('beta'))
with pytest.raises(TypeError):
sapilocal.orang_sample('ising', {}, _params(beta='infinite'))
with pytest.raises(RuntimeError):
sapilocal.orang_sample('ising', {}, _params(beta=-2.5))
with pytest.raises(TypeError):
sapilocal.orang_sample('ising', {}, _params(random_seed='extrarandom'))
def test_bad_problem_type():
with pytest.raises(ValueError):
sapilocal.orang_sample('zzz', {}, VALID_PARAMS)