def check_bloch_normalization(N, epsilon):
Theta, Phi = inv.construct_grid(N)
Angles = np.array([np.cos(Theta), Phi])
Densities = inv.G_angles_q12(Angles, epsilon)
Scaled_densities = Densities*np.sin(Theta)
Total_prob = np.trapz(np.trapz(Scaled_densities, Phi), Theta[:,0])
assert_greater(Total_prob, 0.95)
assert_less_equal(Total_prob, 1)
'given eigenbasis measurements ' +
'in equivalence with Das-Arvind.')
parser.add_argument('-e', '--epsilon', type=float, default=0.575,
help='Strength of the weak measurements')
parser.add_argument('-N', type=int, default=256,
help='Resolution at which to calculate densities')
parser.add_argument('-f', '--folder', type=str, default='../../data/',
help='Folder to write results to')
args = parser.parse_args()
data_file = h5py.File(args.folder + 'bloch_densities_e' +
str(args.epsilon) + '_N' + str(args.N) + '_' +
dt.utcnow().strftime('%Y%m%dT%H%M%SZ') + '.hdf5',
'w-')
Theta, Phi = construct_grid(args.N)
# Radial coordinates for a Lambert azimuthal equal-area projection
R = 2*np.cos(pi/2 - Theta/2)
X = np.sin(Theta) * np.cos(Phi)
Y = np.sin(Theta) * np.sin(Phi)
Z = np.cos(Theta)
Angles = np.array([cos(Theta), Phi])
Densities = G_angles_q12(Angles, args.epsilon)
dataset = data_file.create_dataset('densities', data=Densities)
dataset.attrs['epsilon'] = args.epsilon
dataset.attrs['N'] = args.N
