test: all/1

mimiquate · Nov 1, 2023 · 9ddde1d · 9ddde1d
1 parent 76bc91e
commit 9ddde1d
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/test/candlex_test.exs b/test/candlex_test.exs
@@ -2143,6 +2143,20 @@ defmodule CandlexTest do
       )
     end
 
+    test "all" do
+      t([0, 1, 2])
+      |> Nx.all()
+      |> assert_equal(t(0))
+
+      # t([[-1, 0, 1], [2, 3, 4]], names: [:x, :y])
+      # |> Nx.all(axes: [:x])
+      # |> assert_equal(t([1, 0, 1]))
+
+      # t([[-1, 0, 1], [2, 3, 4]], names: [:x, :y])
+      # |> Nx.all(axes: [:y])
+      # |> assert_equal(t([0, 1]))
+    end
+
     if Candlex.Backend.cuda_available?() do
       test "different devices" do
         t([1, 2, 3], backend: {Candlex.Backend, device: :cpu})