def test_anneal_minimize_calls_trace_func(): trace_func = mock.Mock() anneal.anneal_minimize( 'initial', lambda s: 1.0 if s == 'initial' else 0.0, lambda s: 'better', lambda: 1.0, 1.0, 0.5, 0.5, 1, trace_func=trace_func) trace_func.assert_has_calls([mock.call('initial', 1.0, 1.0, 1.0, True), mock.call('better', 1.0, 0.0, 1.0, True)])
def test_find_sequence_assembles_head_and_tail(): q00 = GridQubit(0, 0) q01 = GridQubit(0, 1) q02 = GridQubit(0, 2) qubits = [q00, q01, q02] start = q01 search = GreedySequenceSearch(_create_device(qubits), start) with mock.patch.object(search, '_sequence_search') as sequence_search: head = [q01, q00] tail = [q01, q02] sequence_search.side_effect = [tail, head] assert search._find_sequence() == qubits sequence_search.assert_has_calls( [mock.call(start, []), mock.call(start, tail)])
def test_find_sequence_assembles_head_and_tail(): q00 = cirq.GridQubit(0, 0) q01 = cirq.GridQubit(0, 1) q02 = cirq.GridQubit(0, 2) qubits = [q00, q01, q02] start = q01 search = greedy.GreedySequenceSearch(_create_device(qubits), start) with mock.patch.object(search, '_sequence_search') as sequence_search: head = [q01, q00] tail = [q01, q02] sequence_search.side_effect = [tail, head] assert search._find_sequence() == qubits sequence_search.assert_has_calls( [mock.call(start, []), mock.call(start, tail)])