mirror of
https://github.com/Cian-H/nanoconc.git
synced 2025-12-22 22:22:01 +00:00
53 lines
2.1 KiB
Julia
53 lines
2.1 KiB
Julia
module TestUtils
|
|
using Serialization
|
|
|
|
using Test
|
|
|
|
function singleton_include(filepath::String, m::Symbol, calling_module::Module)
|
|
try
|
|
target = getproperty(calling_module, m)
|
|
if !isdefined(target, :Module)
|
|
raise("Module $m not defined in $calling_module")
|
|
end
|
|
catch
|
|
Base.include(calling_module, filepath)
|
|
end
|
|
end
|
|
|
|
singleton_include("../anchors.jl", :Anchors, @__MODULE__)
|
|
|
|
import .Anchors: TEST_DIR, SRC_DIR, ROOT_DIR
|
|
|
|
function init_pyenv()
|
|
if "PYTHON" in keys(ENV) && ENV["PYTHON"] != joinpath(ROOT_DIR, ".venv/bin/python")
|
|
run(`$ROOT_DIR/setup_venv.sh`)
|
|
ENV["PYTHON"] = joinpath(ROOT_DIR, ".venv/bin/python")
|
|
end
|
|
end
|
|
|
|
|
|
function fieldvalues(obj)
|
|
[getfield(obj, f) for f in fieldnames(typeof(obj))]
|
|
end
|
|
|
|
deep_compare(a, b; rtol::Real=sqrt(eps()), atol::Real=0.0) = a == b # base case: use ==
|
|
deep_compare(a::Union{AbstractFloat,Complex}, b::Union{AbstractFloat,Complex}; rtol::Real=sqrt(eps()), atol::Real=0.0) = isapprox(a, b) # for floats, use isapprox
|
|
deep_compare(a::Union{Array{AbstractFloat},Array{Complex}}, b::Union{Array{AbstractFloat},Array{Complex}}; rtol::Real=sqrt(eps()), atol::Real=0.0) = isapprox(a, b) # for arrays of floats, use isapprox element-wise
|
|
deep_compare(a::AbstractArray, b::AbstractArray; rtol::Real=sqrt(eps()), atol::Real=0.0) = all(deep_compare.(a, b; rtol, atol)) # for arrays of other types, recurse
|
|
deep_compare(a::Tuple, b::Tuple; rtol::Real=sqrt(eps()), atol::Real=0.0) = all(deep_compare.(a, b; rtol, atol)) # for tuples, recurse
|
|
deep_compare(a::T, b::T; rtol::Real=sqrt(eps()), atol::Real=0.0) where {T<:Any} = deep_compare(fieldvalues(a), fieldvalues(b); rtol, atol) # for composite types, recurse
|
|
|
|
function test_from_serialized(fn::Function, filename::String)
|
|
argskwargs, out = open(filename, "r") do f
|
|
deserialize(f)
|
|
end
|
|
|
|
@test deep_compare([fn(a...; kw...) for (a, kw) in argskwargs], out)
|
|
end
|
|
|
|
function asymmetric_floatvec_to_complexvec(vec_a::Vector{Float32}, vec_b::Vector{Float32})
|
|
shortest = min(length(vec_a), length(vec_b))
|
|
ComplexF32.(vec_a[1:shortest], vec_b[1:shortest])
|
|
end
|
|
|
|
end |