fix test units

demisjohn · May 7, 2022 · 5ed9ac5 · 5ed9ac5
1 parent c407364
commit 5ed9ac5
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/testsuite/metal_coupler.py b/testsuite/metal_coupler.py
@@ -80,7 +80,7 @@ def test_metal_coupler(self):
 
         guided  = 0
         campmax = 0
-        for k in np.arange(0,nom,1):
+        for k in range(0,nom,1):
             camp = abs(waveguide.mode(k).field(Coord(d_sub+dclad+\
                                             guide_thickness/2,0,0)).E1())
             if campmax < camp:
@@ -89,7 +89,7 @@ def test_metal_coupler(self):
 
         # Set input for calculating the fields.
 
-        inc = zeros(N())
+        inc = np.zeros(N())
         inc[guided] = 1
         stack.set_inc_field(inc)
 

diff --git a/testsuite/stack0.py b/testsuite/stack0.py
@@ -30,7 +30,7 @@ def teststack0(self):
 
         s = Stack(AlGaAs(0) + air(1) + AlGaAs(0))
 
-        inc = zeros(N())
+        inc = np.zeros(N())
         inc[0] = 1
         s.set_inc_field(inc)