Support boolean not ! (#150)

adrhill · Aug 7, 2024 · c6d0648 · c6d0648
1 parent 61cb867
commit c6d0648
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 1 deletion.
diff --git a/src/operators.jl b/src/operators.jl
@@ -85,7 +85,7 @@ end
 # ∂²f/∂x² == 0
 ops_1_to_1_z = (
     round, floor, ceil, trunc,
-    sign,
+    sign, !
 )
 for op in ops_1_to_1_z
     T = typeof(op)

diff --git a/test/classification.jl b/test/classification.jl
@@ -43,6 +43,10 @@ random_input(::typeof(sincosd)) = 180 * rand()
 random_first_input(op) = random_input(op)
 random_second_input(op) = random_input(op)
 
+## Skip tests on functions that don't support ForwardDiff's Dual numbers
+
+correct_classification_1_to_1(op::typeof(!), x; atol) = true
+
 ## Derivatives and special cases
 
 both_derivatives_1_to_1(op, x) = derivative(op, x), second_derivative(op, x)

diff --git a/test/test_gradient.jl b/test/test_gradient.jl
@@ -52,6 +52,7 @@ NNLIB_ACTIVATIONS = union(NNLIB_ACTIVATIONS_S, NNLIB_ACTIVATIONS_F)
         x = rand(2)
         g(x) = [x[1] * x[2], ceil(x[1] * x[2]), x[1] * round(x[2])]
         @test jacobian_sparsity(g, x, method) == [1 1; 0 0; 1 0]
+        @test jacobian_sparsity(!, true, method) ≈ [0;;]
 
         # Code coverage
         @test jacobian_sparsity(x -> [sincos(x)...], 1, method) ≈ [1; 1]