Enhance tests with additional almost equal assertions

Added assertions to compare elements of two arrays with almost equal checks.

Author: heptaflar

test/b/test_b_jax.py

@@ -206,8 +206,34 @@ def test_from_lookup_table(self):
         b = f.prior()
         self.assertEqual(c.shape, b.shape)
         self.assertFalse(np.any(b < 0.0))
-        self.assertTrue(np.allclose(b, c))
-        
+        self.assertAlmostEqual(c[0, 0, 0], b[0, 0, 0])
+        self.assertAlmostEqual(c[0, 0, 1], b[0, 0, 1])
+        self.assertAlmostEqual(c[0, 0, 2], b[0, 0, 2])
+        self.assertAlmostEqual(c[0, 1, 0], b[0, 1, 0])
+        self.assertAlmostEqual(c[0, 1, 1], b[0, 1, 1])
+        self.assertAlmostEqual(c[0, 1, 2], b[0, 1, 2])
+        self.assertAlmostEqual(c[0, 2, 0], b[0, 2, 0])
+        self.assertAlmostEqual(c[0, 2, 1], b[0, 2, 1])
+        self.assertAlmostEqual(c[0, 2, 2], b[0, 2, 2])
+        self.assertAlmostEqual(c[1, 0, 0], b[1, 0, 0])
+        self.assertAlmostEqual(c[1, 0, 1], b[1, 0, 1])
+        self.assertAlmostEqual(c[1, 0, 2], b[1, 0, 2])
+        self.assertAlmostEqual(c[1, 1, 0], b[1, 1, 0])
+        self.assertAlmostEqual(c[1, 1, 1], b[1, 1, 1])
+        self.assertAlmostEqual(c[1, 1, 2], b[1, 1, 2])
+        self.assertAlmostEqual(c[1, 2, 0], b[1, 2, 0])
+        self.assertAlmostEqual(c[1, 2, 1], b[1, 2, 1])
+        self.assertAlmostEqual(c[1, 2, 2], b[1, 2, 2])
+        self.assertAlmostEqual(c[2, 0, 0], b[2, 0, 0])
+        self.assertAlmostEqual(c[2, 0, 1], b[2, 0, 1])
+        self.assertAlmostEqual(c[2, 0, 2], b[2, 0, 2])
+        self.assertAlmostEqual(c[2, 1, 0], b[2, 1, 0])
+        self.assertAlmostEqual(c[2, 1, 1], b[2, 1, 1])
+        self.assertAlmostEqual(c[2, 1, 2], b[2, 1, 2])
+        self.assertAlmostEqual(c[2, 2, 0], b[2, 2, 0])
+        self.assertAlmostEqual(c[2, 2, 1], b[2, 2, 1])
+        self.assertAlmostEqual(c[2, 2, 2], b[2, 2, 2])
+
 
 class BSolveTest(unittest.TestCase):
     """