# pylint: disable=wrong-or-nonexistent-copyright-notice
from matplotlib import pyplot as plt
import pytest

from cirq_google.optimizers.two_qubit_gates import example


@pytest.mark.usefixtures('closefigures')
def test_gate_compilation_example():
    plt.switch_backend('agg')
    example.main(samples=10, max_infidelity=0.3)

# pylint: disable=wrong-or-nonexistent-copyright-notice
from matplotlib import pyplot as plt
import pytest

from cirq_google.optimizers.two_qubit_gates import example


@pytest.mark.usefixtures('closefigures')
def test_gate_compilation_example():
    plt.switch_backend('agg')
    example.main(samples=10, max_infidelity=0.3)

# pylint: disable=wrong-or-nonexistent-copyright-notice