Loop over different indices in 2D (F)FB tests

Author: janden

tests/test_FBbasis2D.py

@@ -38,10 +38,9 @@ def setUp(self):
     def tearDown(self):
         pass
 
-    def testElement(self):
-        ell = 1
-        sgn = -1
-        k = 2
+    def _testElement(self, ell, k, sgn):
+        # This is covered by the isotropic test.
+        assert ell > 0
 
         indices = self.basis.indices()
         ells = indices["ells"]
@@ -74,6 +73,14 @@ def testElement(self):
         self.assertTrue(np.allclose(im, im_ref, atol=1e-4))
         self.assertTrue(np.allclose(coef, coef_ref, atol=1e-4))
 
+    def testElements(self):
+        ells = [1, 1, 1, 1]
+        ks = [1, 2, 1, 2]
+        sgns = [-1, -1, 1, 1]
+
+        for ell, k, sgn in zip(ells, ks, sgns):
+            self._testElement(ell, k, sgn)
+
     def testComplexCoversion(self):
         x = randn(*self.basis.sz, seed=self.seed)
 

tests/test_FFBbasis2D.py

@@ -42,11 +42,7 @@ def setUp(self):
     def tearDown(self):
         pass
 
-    def testElement(self):
-        ell = 1
-        sgn = -1
-        k = 2
-
+    def _testElement(self, ell, k, sgn):
         indices = self.basis.indices()
         ells = indices["ells"]
         sgns = indices["sgns"]
@@ -91,6 +87,14 @@ def testElement(self):
         self.assertTrue(np.allclose(im, im_ref, atol=1e-1))
         self.assertTrue(np.allclose(coef, coef_ref, atol=1e-1))
 
+    def testElements(self):
+        ells = [1, 1, 1, 1]
+        ks = [1, 2, 1, 2]
+        sgns = [-1, -1, 1, 1]
+
+        for ell, k, sgn in zip(ells, ks, sgns):
+            self._testElement(ell, k, sgn)
+
     def testRotate(self):
         # Convergence issues for double precision.
         if np.dtype(self.dtype) is np.dtype(np.float64):