debuging tests

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

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/test/ex_vision/style_transfer/style_transfer_test.exs b/test/ex_vision/style_transfer/style_transfer_test.exs
@@ -43,13 +43,23 @@ 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()
 
+      diff_sum =
+        expected_result
+        |> Nx.subtract(result)
+        |> Nx.reduce(0, fn x, y -> Nx.add(x, y) end)
+        |> Nx.to_number()
+
+      assert diff_sum == 1
+
       assert_tensors_equal(result, expected_result)
     end
   end

diff --git a/test/support/exvision/test_utils.ex b/test/support/exvision/test_utils.ex
@@ -1,4 +1,5 @@
 defmodule ExVision.TestUtils do
+  require Nx
   @moduledoc false
 
   import ExUnit.Assertions, only: :macros