add specific tests for BitString and SortedParticleList

joachimbrand · Sep 18, 2023 · e322566 · e322566
1 parent 2aea78b
commit e322566
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/src/BitStringAddresses/sortedparticlelist.jl b/src/BitStringAddresses/sortedparticlelist.jl
@@ -44,6 +44,15 @@ function SortedParticleList{N,M}() where {N,M}
     T = select_int_type(M)
     return SortedParticleList{N,M,T}(ones(SVector{N,T}))
 end
+function SortedParticleList(onr)
+    N = sum(onr)
+    M = length(onr)
+    T = select_int_type(M)
+    eltype(onr) <: Integer || throw(ArgumentError("the onr can only contain integers"))
+    any(≥(0), onr) || throw(ArgumentError("all elements of the onr should be positive"))
+
+    return from_onr(SortedParticleList{N,M,T}, onr)
+end
 function from_onr(::Type{S}, onr) where {N,M,T,S<:SortedParticleList{N,M,T}}
     spl = zeros(MVector{N,T})
     curr = 1

diff --git a/test/RimuIO.jl b/test/RimuIO.jl
@@ -58,6 +58,8 @@ end
 end
 @testset "Addresses" begin
     for addr in (
+        BitString{10}(0b1100110011),
+        SortedParticleList((1, 0, 1, 0, 0, 2, 3)),
         near_uniform(BoseFS{10, 10}),
         BoseFS(101, 5 => 10),
         FermiFS((1,1,1,0,0,0)),