Implemented Arduino-based polarity control into main script

One_Unit_Test.py

@@ -14,42 +14,53 @@ Test4 = 0
 # Constants:
 g.Coil_const = np.array([38.6, 38.45, 37.9])*1e-9  # Coil constants [x,y,z] in T/A
 g.ambientField = np.array([80])*1e-6  # ambient magnetic field in measurement area, to be cancelled out
+g.resistances = np.array([3.9, 1, 1])  # resistance of [x,y,z] circuits
+g.maxWatts = np.array([8, 0, 0])  # max. allowed power for [x,y,z] circuits
 
 # COM-Ports for power supply units:
-xPort = "COM7"  # placeholder
-g.xDevice = PS2000B.PS2000B(xPort)
+xyPort = "COM7"
+g.xyDevice = PS2000B.PS2000B(xyPort)
 
-func.print_status(g.xDevice)
-t.sleep(3)
+g.maxAmps = np.sqrt(g.maxWatts/g.resistances)
+print(g.maxAmps)
 
-g.xDevice.enable_all()
-func.print_status(g.xDevice)
+
+def print_status():
+    print("Output 1:")
+    func.print_status(g.xAxis)
+    print("Output 2:")
+    func.print_status(g.yAxis)
+
+
+g.xyDevice.enable_all()
+func.set_to_zero(g.xyDevice)
+print_status()
 t.sleep(3)
 
 if Test1 == 1:
-    g.xDevice.voltage1 = 5
+    g.xyDevice.voltage1 = 5
     t.sleep(1)
-    func.print_status(g.xDevice)
+    print_status()
     t.sleep(5)
-    func.set_to_zero(g.xDevice)
+    func.set_to_zero(g.xyDevice)
 
 if Test2 == 1:
-    g.xDevice.current1 = 0.2
+    g.xyDevice.current1 = 0.2
     t.sleep(1)
-    func.print_status(g.xDevice)
+    print_status()
     t.sleep(5)
-    func.set_to_zero(g.xDevice)
+    func.set_to_zero(g.xyDevice)
 
 if Test4 == 1:
-    func.set_axis_current(g.xDevice, 0.2)
+    func.set_axis_current(g.xyDevice, 0.2)
     t.sleep(1)
-    func.print_status(g.xDevice)
+    print_status()
     t.sleep(10)
-    func.set_to_zero(g.xDevice)
+    func.set_to_zero(g.xyDevice)
     t.sleep(1)
 
-func.print_status(g.xDevice)
+print_status()
 
-g.xDevice.disable_all()
+g.xyDevice.disable_all()
 
-func.print_status(g.xDevice)
+print_status()