test: test cover Nx.logical_not

mimiquate · Nov 6, 2023 · ca8e83f · ca8e83f
1 parent bbe4bdc
commit ca8e83f
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/test/candlex_test.exs b/test/candlex_test.exs
@@ -1192,6 +1192,16 @@ defmodule CandlexTest do
       )
     end
 
+    test "logical_not" do
+      t([-1, 0, 1])
+      |> Nx.logical_not()
+      |> assert_equal(t([0, 1, 0]))
+
+      t([-1.0, 0.0, 1.0])
+      |> Nx.logical_not()
+      |> assert_equal(t([0, 1, 0]))
+    end
+
     test "erf" do
       Nx.erf(1.0)
       |> assert_close(t(0.8427007794380188))