use shouldBeEps

Author: chessai

test/ArrayFire/ArithSpec.hs

@@ -22,23 +22,23 @@ spec =
     it "Should subtract two scalar arrays" $ do
       scalar @Int 4 - 2 `shouldBe` 2
     it "Should multiply two scalar arrays" $ do
-      scalar @Double 4 `mul` 2 `shouldBe` 8
+      scalar @Double 4 `mul` 2 `shouldBeEps` 8
     it "Should divide two scalar arrays" $ do
-      div @Double 8 2 `shouldBe` 4
+      div @Double 8 2 `shouldBeEps` 4
     it "Should add two matrices" $ do
       matrix @Int (2,2) [[1,1],[1,1]] + matrix @Int (2,2) [[1,1],[1,1]]
         `shouldBe`
            matrix @Int (2,2) [[2,2],[2,2]]
     it "Should take cubed root" $ do
-      3 `shouldBe` cbrt @Double 27
+      3 `shouldBeEps` cbrt @Double 27
     it "Should take square root" $ do
-      2 `shouldBe` sqrt @Double 4
+      2 `shouldBeEps` sqrt @Double 4
     it "Should lt Array" $ do
-      2 `A.lt` (3 :: Array Double) `shouldBe` True
+      2 `A.lt` (3 :: Array Double) `shouldBeEps` True
     it "Should gt Array" $ do
-      2 `A.gt` (3 :: Array Double) `shouldBe` False
+      2 `A.gt` (3 :: Array Double) `shouldBeEps` False
     it "Should eq Array" $ do
-      3 == (3 :: Array Double) `shouldBe` True
+      3 == (3 :: Array Double) `shouldBeEps` True
     it "Should and Array" $ do
       (mkArray @CBool [1] [0] `and` mkArray [1] [1])
          `shouldBe` mkArray [1] [0]
@@ -92,3 +92,12 @@ spec =
       isZero (scalar @Double (acos 2)) `shouldBe` scalar @Double 0
       isZero (scalar @Double 0) `shouldBe` scalar @Double 1
       isZero (scalar @Double 1) `shouldBe` scalar @Double 0
+
+shouldBeEps :: Array Double -> Array Double -> Expectation
+actual `shouldBeEps` expected = expect err_msg (cmpEps actual expected)
+  where
+    err_msg = "expected: " ++ show expected ++ "\n but got: " ++ show actual)
+    cmpEps =
+      let x :: Double
+          x = fromIntegral $ abs $ actual - expected
+      in x <= 1e-14