diff --git a/test/candlex_test.exs b/test/candlex_test.exs
index 111e86f..31c5ad2 100644
--- 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})