Skip to content

Commit

Permalink
test: make complex test harder
Browse files Browse the repository at this point in the history
  • Loading branch information
MilesCranmer committed Dec 26, 2024
1 parent c07d598 commit cd44268
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion test/test_parametric_template_expressions.jl
Original file line number Diff line number Diff line change
Expand Up @@ -306,6 +306,7 @@ end
subex_f32 = ComposableExpression(
Node{ComplexF32}(; feature=1); operators=Options().operators, variable_names=["x"]
)
subex_f32 = 2.0 * subex_f32
param32 = ComplexF32[10.0 + 0im, 20.0 + 0im]

expr_f32 = TemplateExpression(
Expand All @@ -318,6 +319,6 @@ end
@test eltype(get_metadata(expr_f32).parameters._data) == ComplexF32

Xtest = reshape(ComplexF32[2.0 + 0im], 1, 1)
@test expr_f32(Xtest) == [ComplexF32(32.0 + 0im)]
@test expr_f32(Xtest) == [ComplexF32(34.0 + 0im)]
@test expr_f32(Xtest) isa Vector{ComplexF32}
end

0 comments on commit cd44268

Please sign in to comment.