def METHOD_NAME(self):
    """There is a known bug in the qiskit backend for azure where if the shots
    argument might be ignored. This test checks that the output from the azure
    computation matches the input requirements.
    """
    shots = 1024
    problem_qubo = NumberPartition([1, 2, 3]).qubo
    azure_device = create_device(
        location="azure",
        name="rigetti.sim.qvm",
        resource_id=self.RESOURCE_ID,
        az_location=self.AZ_LOCATION,
    )
    q = QAOA()
    q.set_device(azure_device)
    q.set_backend_properties(n_shots=shots)
    q.set_classical_optimizer(maxiter=1)
    q.compile(problem_qubo)
    q.optimize()
    comp_shots = sum(q.result.optimized["optimized measurement outcomes"].values())
    self.assertEqual(shots, comp_shots)