added getB vector input tests, removed bad test

Author: OrtnerMichael

tests/test_Collection.py

@@ -243,35 +243,6 @@ def test_GetB():
         assert round(result[j],rounding) == round(mockResult[j],rounding)
 
 
-def test_GetBList():
-    # Check if sole getB throws an error
-    # for Lists.
-    from magpylib._lib.classes.magnets import Box
-    #Input
-    mockList = (    array([0.12488298, 0.10927261, 0.07805186]),
-                    array([0.12488298, 0.10927261, 0.07805186]),)
-    mockResult = [  sum(mockList),
-                    sum(mockList)]
-
-    mag=(2,3,5)
-    dim=(2,2,2)
-    pos=array([   [2,2,2],
-                [2,2,2]])
-
-
-    with pytest.raises(IndexError):
-    #Run   
-        b = Box(mag,dim)
-        b2 = Box(mag,dim)
-        c = Collection(b,b2)
-        result = c.getB(pos) 
-
-        rounding = 4
-        for i in range(len(mockResult)):
-            for j in range(3):
-                assert round(result[i][j],rounding) == round(mockResult[i][j],rounding)
-
-
 def test_AddList():
     # Check if adding a list of generics to collection 
     # does not throw an error

tests/test_sources/test_getB_vectorInput.py

@@ -0,0 +1,42 @@
+import numpy as np
+import time
+import magpylib as magpy
+
+POS = np.array([[1,2,3],[2,-3,4],[-3,4,5],[5,6,-7],[-3,-2,1],[-4,3,-2],[5,-4,-3],[-6,-5,-4]])
+
+s = magpy.source.current.Circular(curr=123,dim=2,pos=(2,-4,5),angle=23,axis=(.2,-5,3))
+B = np.array([s.getB(p) for p in POS])
+Bv = s.getB(POS)
+err = np.amax(np.abs(Bv-B))
+assert err < 1e-12
+
+VERT = POS*.333
+s = magpy.source.current.Line(curr=123,vertices=VERT)
+B = np.array([s.getB(p) for p in POS])
+Bv = s.getB(POS)
+err = np.amax(np.abs(Bv-B))
+assert err < 1e-12
+
+s = magpy.source.moment.Dipole(moment=(.1,-5,5.5),pos=(2,-4,5),angle=33,axis=(.2,-5,3))
+B = np.array([s.getB(p) for p in POS])
+Bv = s.getB(POS)
+err = np.amax(np.abs(Bv-B))
+assert err < 1e-12
+
+s = magpy.source.magnet.Box(mag=(33,22,111),dim=(3,2,1),pos=(2,-4,5),angle=33,axis=(.2,-5,3))
+B = np.array([s.getB(p) for p in POS])
+Bv = s.getB(POS)
+err = np.amax(np.abs(Bv-B))
+assert err < 1e-12
+
+s = magpy.source.magnet.Cylinder(mag=(33,22,111),dim=(3,1),pos=(2,-4,5),angle=33,axis=(.2,-5,3))
+B = np.array([s.getB(p) for p in POS])
+Bv = s.getB(POS)
+err = np.amax(np.abs(Bv-B))
+assert err < 1e-12
+
+s = magpy.source.magnet.Sphere(mag=(33,22,111),dim=.123,pos=(2,-4,5),angle=33,axis=(.2,-5,3))
+B = np.array([s.getB(p) for p in POS])
+Bv = s.getB(POS)
+err = np.amax(np.abs(Bv-B))
+assert err < 1e-12