




import math

import numpy as np
import qiskit
from qiskit import Aer, ClassicalRegister, QuantumCircuit, QuantumRegister, execute





def __UpperCamelCase			(       _A = 3  ):
						if isinstance(_A ,  _A  ):
												raise TypeError('''number of qubits must be a integer.'''  )
						if number_of_qubits <= 0:
												raise ValueError('''number of qubits must be > 0.'''  )
						if math.floor(_A  ) != number_of_qubits:
												raise ValueError('''number of qubits must be exact integer.'''  )
						if number_of_qubits > 10:
												raise ValueError('''number of qubits too large to simulate(>10).'''  )

						lowerCAmelCase_      			=  QuantumRegister(_A ,  '''qr'''  )
						lowerCAmelCase_      			=  ClassicalRegister(_A ,  '''cr'''  )

						lowerCAmelCase_      			=  QuantumCircuit(_A ,  _A  )

						lowerCAmelCase_      			=  number_of_qubits

						for i in range(_A  ):
												quantum_circuit.h(number_of_qubits - i - 1  )
												counter -= 1
												for j in range(_A  ):
																		quantum_circuit.cp(np.pi / 2 ** (counter - j) ,  _A ,  _A  )

						for k in range(number_of_qubits // 2  ):
												quantum_circuit.swap(_A ,  number_of_qubits - k - 1  )

						# measure all the qubits
						quantum_circuit.measure(_A ,  _A  )
						# simulate with 10000 shots
						lowerCAmelCase_      			=  Aer.get_backend('''qasm_simulator'''  )
						lowerCAmelCase_      			=  execute(_A ,  _A ,  shots=10000  )

						return job.result().get_counts(_A  )


if __name__ == "__main__":
						print(
						    f"Total count for quantum fourier transform state is: \
    {quantum_fourier_transform(3)}"
						)










import unittest

import numpy as np
import torch

from diffusers import KarrasVePipeline, KarrasVeScheduler, UNetaDModel
from diffusers.utils.testing_utils import enable_full_determinism, require_torch, slow, torch_device


enable_full_determinism()


class  A		(  unittest.TestCase     ):



		@property
		def        SCREAMING_SNAKE_CASE__			(     self   ):







								"""simple docstring"""



								torch.manual_seed(0   )
								lowerCAmelCase_      			=  UNetaDModel(
								    block_out_channels=(32, 64),	layers_per_block=2,	sample_size=32,	in_channels=3,	out_channels=3,	down_block_types=('''DownBlock2D''', '''AttnDownBlock2D'''),	up_block_types=('''AttnUpBlock2D''', '''UpBlock2D'''),	)
								return model







		def        SCREAMING_SNAKE_CASE__			(     self   ):







								"""simple docstring"""



								lowerCAmelCase_      			=  self.dummy_uncond_unet
								lowerCAmelCase_      			=  KarrasVeScheduler()

								lowerCAmelCase_      			=  KarrasVePipeline(unet=UpperCamelCase__,	scheduler=UpperCamelCase__   )
								pipe.to(UpperCamelCase__   )
								pipe.set_progress_bar_config(disable=UpperCamelCase__   )

								lowerCAmelCase_      			=  torch.manual_seed(0   )
								lowerCAmelCase_      			=  pipe(num_inference_steps=2,	generator=UpperCamelCase__,	output_type='''numpy'''   ).images

								lowerCAmelCase_      			=  torch.manual_seed(0   )
								lowerCAmelCase_      			=  pipe(num_inference_steps=2,	generator=UpperCamelCase__,	output_type='''numpy''',	return_dict=UpperCamelCase__   )[0]

								lowerCAmelCase_      			=  image[0, -3:, -3:, -1]
								lowerCAmelCase_      			=  image_from_tuple[0, -3:, -3:, -1]

								assert image.shape == (1, 32, 32, 3)
								lowerCAmelCase_      			=  np.array([0.0, 1.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0]   )

								assert np.abs(image_slice.flatten() - expected_slice   ).max() < 1E-2
								assert np.abs(image_from_tuple_slice.flatten() - expected_slice   ).max() < 1E-2







@slow
@require_torch
class  A		(  unittest.TestCase     ):



		def        SCREAMING_SNAKE_CASE__			(     self   ):







								"""simple docstring"""



								lowerCAmelCase_      			=  '''google/ncsnpp-celebahq-256'''
								lowerCAmelCase_      			=  UNetaDModel.from_pretrained(UpperCamelCase__   )
								lowerCAmelCase_      			=  KarrasVeScheduler()

								lowerCAmelCase_      			=  KarrasVePipeline(unet=UpperCamelCase__,	scheduler=UpperCamelCase__   )
								pipe.to(UpperCamelCase__   )
								pipe.set_progress_bar_config(disable=UpperCamelCase__   )

								lowerCAmelCase_      			=  torch.manual_seed(0   )
								lowerCAmelCase_      			=  pipe(num_inference_steps=20,	generator=UpperCamelCase__,	output_type='''numpy'''   ).images

								lowerCAmelCase_      			=  image[0, -3:, -3:, -1]
								assert image.shape == (1, 256, 256, 3)
								lowerCAmelCase_      			=  np.array([0.578, 0.5_811, 0.5_924, 0.5_809, 0.587, 0.5_886, 0.5_861, 0.5_802, 0.586]   )

								assert np.abs(image_slice.flatten() - expected_slice   ).max() < 1E-2




