Merge pull request #20 from theogf/patch-1

Add kwarg constructor for TupleVector
cscherrer · Dec 15, 2021 · d709776 · d709776 · cscherrer · Dec 15, 2021
2 parents 4713518 + 8490cc7
commit d709776
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 1 deletion.
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "TupleVectors"
 uuid = "615932cf-77b6-4358-adcd-5b7eba981d7e"
 authors = ["Chad Scherrer <[email protected]> and contributors"]
-version = "0.1.3"
+version = "0.1.4"
 
 [deps]
 ArraysOfArrays = "65a8f2f4-9b39-5baf-92e2-a9cc46fdf018"

diff --git a/src/tuplevector.jl b/src/tuplevector.jl
@@ -36,6 +36,10 @@ function TupleVector(::UndefInitializer, x::T, n::Int) where {T<:NamedTuple}
     return TupleVector{T, typeof(data)}(data)
 end
 
+function TupleVector(; kwargs...)
+    return TupleVector(NamedTuple(kwargs)) 
+end
+
 # function TupleVector(x::Union{Tuple, NamedTuple})
 #     flattened = flatten(x)
 #     @assert allequal(size.(flattened)...)

diff --git a/test/runtests.jl b/test/runtests.jl
@@ -6,6 +6,14 @@ using MeasureTheory
 using LinearAlgebra
 
 @testset "TupleVectors.jl" begin
+    ## Testing constructors
+    w = randn(10)
+    y = randn(10)
+    nt = (w=w, y=y)
+    A = TupleVector(nt)
+    B = TupleVector(; w=w, y=y)
+    @test A.w == B.w
+
     @test chainvec(3,5)[1] == 3