add propertynames

BeastyBlacksmith · Jun 17, 2019 · 4fa3c52 · 4fa3c52 · BeastyBlacksmith · Jun 17, 2019
1 parent b507581
commit 4fa3c52
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/src/ProtoStruct.jl b/src/ProtoStruct.jl
@@ -1,13 +1,17 @@
 macro proto( name )
  ex = quote
- struct $name{ NT<:NamedTuple{Syms,T} where {Syms,T} }
+ struct $name{NT<:NamedTuple}
  properties::NT
  end # struct
 
  $name(;kwargs...) = $name(kwargs.data)
 
  function Base.getproperty( o::$name, s::Symbol )
- return getproperty( getfield(o,:properties), s )
+ return getproperty( getfield(o, :properties), s )
+ end # function
+
+ function Base.propertynames( o::$name )
+ return propertynames( getfield(o, :properties) )
  end # function
  end # quote
  esc(ex)

diff --git a/test/test_ProtoStruct.jl b/test/test_ProtoStruct.jl
@@ -13,3 +13,7 @@ end # testset
  @test test_me.B == "2"
  @test test_me.C == complex(1)
 end # testset
+
+@testset "Properties" begin
+ @test propertynames( test_me ) == (:A, :B, :C)
+end # testset