interferometer 1.1__tar.gz → 1.1.1__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: interferometer
-Version: 1.1
+Version: 1.1.1
 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,8 @@ Classifier: Topic :: Scientific/Engineering :: Physics
 Requires-Python: >=3.7
 Description-Content-Type: text/markdown
 License-File: LICENSE
+Requires-Dist: numpy
+Requires-Dist: matplotlib
 
 # Interferometer package
 

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

@@ -185,8 +185,8 @@ 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])
+            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.complex_)
             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)
@@ -216,8 +216,8 @@ 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])
+                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.complex_)
                 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)
@@ -228,8 +228,8 @@ 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])
+                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.complex_)
                 T[modes[0]-1, modes[0]-1] = np.exp(1j * phi) * np.cos(theta)
                 T[modes[0]-1, modes[1]-1] = -np.sin(theta)
@@ -246,8 +246,8 @@ def square_decomposition(U):
         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)
@@ -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.1}/interferometer.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: interferometer
-Version: 1.1
+Version: 1.1.1
 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,8 @@ Classifier: Topic :: Scientific/Engineering :: Physics
 Requires-Python: >=3.7
 Description-Content-Type: text/markdown
 License-File: LICENSE
+Requires-Dist: numpy
+Requires-Dist: matplotlib
 
 # Interferometer package
 

{interferometer-1.1 → interferometer-1.1.1}/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.1}/pyproject.toml

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

interferometer-1.1.1/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)
+