interferometer 1.1__tar.gz → 1.1.2__tar.gz

{interferometer-1.1 → interferometer-1.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.4
 Name: interferometer
-Version: 1.1
+Version: 1.1.2
 Summary: Algorithms for universal interferometers
 Author-email: "William R. Clements" <mail@william-clements.com>
 Project-URL: Homepage, https://github.com/clementsw/interferometer
@@ -13,6 +13,9 @@ Classifier: Topic :: Scientific/Engineering :: Physics
 Requires-Python: >=3.7
 Description-Content-Type: text/markdown
 License-File: LICENSE
+Requires-Dist: numpy
+Requires-Dist: matplotlib
+Dynamic: license-file
 
 # Interferometer package
 

{interferometer-1.1 → interferometer-1.1.2}/interferometer/main.py

@@ -84,10 +84,10 @@ class Interferometer:
             complex-valued 2D numpy array representing the interferometer
         """
         N = int(self.count_modes())
-        U = np.eye(N, dtype=np.complex_)
+        U = np.eye(N, dtype=np.complex128)
 
         for BS in self.BS_list:
-            T = np.eye(N, dtype=np.complex_)
+            T = np.eye(N, dtype=np.complex128)
             T[BS.mode1 - 1, BS.mode1 - 1] = np.exp(1j * BS.phi) * np.cos(BS.theta)
             T[BS.mode1 - 1, BS.mode2 - 1] = -np.sin(BS.theta)
             T[BS.mode2 - 1, BS.mode1 - 1] = np.exp(1j * BS.phi) * np.sin(BS.theta)
@@ -185,9 +185,9 @@ def triangle_decomposition(U):
     for ii in range(N-1):
         for jj in range(N-1-ii):
             modes = [N - jj - 1, N - jj]
-            theta = np.arctan(abs(U[ii, N - 1 - jj] / U[ii, N - 2 - jj]))
-            phi = -np.angle(-U[ii, N - 1 - jj] / U[ii, N - 2 - jj])
-            invT = np.eye(N, dtype=np.complex_)
+            theta = custom_arctan(U[ii, N - 1 - jj], U[ii, N - 2 - jj])
+            phi = -custom_angle(-U[ii, N - 1 - jj], U[ii, N - 2 - jj])
+            invT = np.eye(N, dtype=np.complex128)
             invT[modes[0]-1, modes[0]-1] = np.exp(-1j * phi) * np.cos(theta)
             invT[modes[0]-1, modes[1]-1] = np.exp(-1j * phi) * np.sin(theta)
             invT[modes[1]-1, modes[0]-1] = -np.sin(theta)
@@ -216,9 +216,9 @@ def square_decomposition(U):
         if np.mod(ii, 2) == 0:
             for jj in range(ii+1):
                 modes = [ii - jj + 1, ii + 2 - jj]
-                theta = np.arctan(abs(U[N-1-jj, ii-jj] / U[N-1-jj, ii-jj+1]))
-                phi = np.angle(U[N-1-jj, ii-jj] / U[N-1-jj, ii-jj+1])
-                invT = np.eye(N, dtype=np.complex_)
+                theta = custom_arctan(U[N-1-jj, ii-jj], U[N-1-jj, ii-jj+1])
+                phi = custom_angle(U[N-1-jj, ii-jj], U[N-1-jj, ii-jj+1])
+                invT = np.eye(N, dtype=np.complex128)
                 invT[modes[0]-1, modes[0]-1] = np.exp(-1j * phi) * np.cos(theta)
                 invT[modes[0]-1, modes[1]-1] = np.exp(-1j * phi) * np.sin(theta)
                 invT[modes[1]-1, modes[0]-1] = -np.sin(theta)
@@ -228,9 +228,9 @@ def square_decomposition(U):
         else:
             for jj in range(ii+1):
                 modes = [N+jj-ii-1, N+jj-ii]
-                theta = np.arctan(abs(U[N+jj-ii-1, jj] / U[N+jj-ii-2, jj]))
-                phi = np.angle(-U[N+jj-ii-1, jj] / U[N+jj-ii-2, jj])
-                T = np.eye(N, dtype=np.complex_)
+                theta = custom_arctan(U[N+jj-ii-1, jj], U[N+jj-ii-2, jj])
+                phi = custom_angle(-U[N+jj-ii-1, jj], U[N+jj-ii-2, jj])
+                T = np.eye(N, dtype=np.complex128)
                 T[modes[0]-1, modes[0]-1] = np.exp(1j * phi) * np.cos(theta)
                 T[modes[0]-1, modes[1]-1] = -np.sin(theta)
                 T[modes[1]-1, modes[0]-1] = np.exp(1j * phi) * np.sin(theta)
@@ -240,14 +240,14 @@ def square_decomposition(U):
 
     for BS in np.flip(left_T, 0):
         modes = [int(BS.mode1), int(BS.mode2)]
-        invT = np.eye(N, dtype=np.complex_)
+        invT = np.eye(N, dtype=np.complex128)
         invT[modes[0]-1, modes[0]-1] = np.exp(-1j * BS.phi) * np.cos(BS.theta)
         invT[modes[0]-1, modes[1]-1] = np.exp(-1j * BS.phi) * np.sin(BS.theta)
         invT[modes[1]-1, modes[0]-1] = -np.sin(BS.theta)
         invT[modes[1]-1, modes[1]-1] = np.cos(BS.theta)
         U = np.matmul(invT, U)
-        theta = np.arctan(abs(U[modes[1]-1, modes[0]-1]/U[modes[1]-1, modes[1]-1]))
-        phi = np.angle(U[modes[1]-1, modes[0]-1] / U[modes[1]-1, modes[1]-1])
+        theta = custom_arctan(U[modes[1]-1, modes[0]-1], U[modes[1]-1, modes[1]-1])
+        phi = custom_angle(U[modes[1]-1, modes[0]-1], U[modes[1]-1, modes[1]-1])
         invT[modes[0]-1, modes[0]-1] = np.exp(-1j * phi) * np.cos(theta)
         invT[modes[0]-1, modes[1]-1] = np.exp(-1j * phi) * np.sin(theta)
         invT[modes[1]-1, modes[0]-1] = -np.sin(theta)
@@ -271,7 +271,7 @@ def random_unitary(N):
     Returns:
         complex-valued 2D numpy array representing the interferometer
     """
-    X = np.zeros([N, N], dtype=np.complex_)
+    X = np.zeros([N, N], dtype=np.complex128)
     for ii in range(N):
         for jj in range(N):
             X[ii, jj] = (np.random.normal() + 1j * np.random.normal()) / np.sqrt(2)
@@ -281,3 +281,15 @@ def random_unitary(N):
     U = np.matmul(q, r)
 
     return U
+
+def custom_arctan(x1, x2):
+    if x2 != 0:
+        return np.arctan(abs(x1/x2))
+    else:
+        return np.pi/2
+
+def custom_angle(x1, x2):
+    if x2 != 0:
+        return np.angle(x1/x2)
+    else:
+        return 0

{interferometer-1.1 → interferometer-1.1.2}/interferometer.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.4
 Name: interferometer
-Version: 1.1
+Version: 1.1.2
 Summary: Algorithms for universal interferometers
 Author-email: "William R. Clements" <mail@william-clements.com>
 Project-URL: Homepage, https://github.com/clementsw/interferometer
@@ -13,6 +13,9 @@ Classifier: Topic :: Scientific/Engineering :: Physics
 Requires-Python: >=3.7
 Description-Content-Type: text/markdown
 License-File: LICENSE
+Requires-Dist: numpy
+Requires-Dist: matplotlib
+Dynamic: license-file
 
 # Interferometer package
 

{interferometer-1.1 → interferometer-1.1.2}/interferometer.egg-info/SOURCES.txt

@@ -7,4 +7,5 @@ interferometer.egg-info/PKG-INFO
 interferometer.egg-info/SOURCES.txt
 interferometer.egg-info/dependency_links.txt
 interferometer.egg-info/requires.txt
-interferometer.egg-info/top_level.txt
+interferometer.egg-info/top_level.txt
+tests/test_interferometer.py

{interferometer-1.1 → interferometer-1.1.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "interferometer"
-version = "1.1"
+version = "1.1.2"
 authors = [
   { name="William R. Clements", email="mail@william-clements.com" },
 ]

interferometer-1.1.2/tests/test_interferometer.py

@@ -0,0 +1,17 @@
+import numpy as np
+from unittest import TestCase
+
+from interferometer import random_unitary, triangle_decomposition, square_decomposition
+
+class TestInterferometer(TestCase):
+
+    def test_triangle_interferometer(self):
+        U = random_unitary(5)
+        I = triangle_decomposition(U)
+        self.assertTrue(abs(np.max(I.calculate_transformation() - U)) < 1e-14)
+
+    def test_square_interferometer(self):
+        U = random_unitary(5)
+        I = square_decomposition(U)
+        self.assertTrue(abs(np.max(I.calculate_transformation() - U)) < 1e-14)
+