debuging tests

software-mansion-labs · Aug 20, 2024 · 568096c · 568096c
1 parent fdb204a
commit 568096c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/test/ex_vision/style_transfer/style_transfer_test.exs b/test/ex_vision/style_transfer/style_transfer_test.exs
@@ -43,14 +43,16 @@ for {module, opts} <- TestConfiguration.configuration() do
     use ExVision.Model.Case, module: unquote(opts[:module])
     use ExVision.TestUtils
 
+    require Logger
+
     @impl true
     def test_inference_result(result) do
       expected_result =
         "test/assets/results/style_transfer/#{unquote(opts[:gt_file])}"
         |> File.read!()
         |> Nx.deserialize()
 
-      assert_tensors_equal(result, expected_result)
+      assert_tensors_equal(result, expected_result, 2, 0.03)
     end
   end
 end
diff --git a/test/support/exvision/test_utils.ex b/test/support/exvision/test_utils.ex
@@ -42,10 +42,10 @@ defmodule ExVision.TestUtils do
     end
   end
 
-  defmacro assert_tensors_equal(a, b, delta \\ @default_delta) do
+  defmacro assert_tensors_equal(a, b, delta \\ @default_delta, relative_delta \\ 0.0) do
     quote do
       assert unquote(a)
-             |> Nx.all_close(unquote(b), atol: unquote(delta))
+             |> Nx.all_close(unquote(b), atol: unquote(delta), rtol: unquote(relative_delta))
              |> Nx.reduce_min()
              |> Nx.to_number() == 1
     end