test: that extract_gradient needs to be defined

MilesCranmer · MilesCranmer · commit d976098da547 · 2024-06-29T21:15:48.000+01:00
diff --git a/test/test_multi_expression.jl b/test/test_multi_expression.jl
@@ -67,6 +67,9 @@
         @test_throws "`set_constants!` function must be implemented for" set_constants!(
             multi_ex, nothing, nothing
         )
+        @test_throws "`extract_gradient` function must be implemented for" extract_gradient(
+            nothing, multi_ex
+        )
     end
 
     tree_factory(f::F, trees) where {F} = f(; trees...)

Original file line number	Diff line number	Diff line change
`@@ -67,6 +67,9 @@`
`67`	`67`	@test_throws "`set_constants!` function must be implemented for" set_constants!(
`68`	`68`	`multi_ex, nothing, nothing`
`69`	`69`	`)`
	`70`	+ @test_throws "`extract_gradient` function must be implemented for" extract_gradient(
	`71`	`+ nothing, multi_ex`
	`72`	`+ )`
`70`	`73`	`end`
`71`	`74`
`72`	`75`	`tree_factory(f::F, trees) where {F} = f(; trees...)`