Add getproperty method

JuliaEarth · Jul 15, 2020 · faeaff2 · faeaff2
1 parent 45882ab
commit faeaff2
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 11 deletions.
diff --git a/src/composition.jl b/src/composition.jl
@@ -102,17 +102,21 @@ Names of parts in the composition `c`.
 """
 names(c::Composition{D,SYMS}) where {D,SYMS} = SYMS
 
-# """
-#     getproperty(c, name)
-#
-# Return the value of part with given `name` in the composition `c`.
-# """
-# function getproperty(c::Composition{D,SYMS}, S::Symbol) where {D,SYMS}
-#   for (i, SYM) in enumerate(SYMS)
-#     S == SYM && (return c.parts[i])
-#   end
-#   @error "invalid part $S"
-# end
+"""
+    getproperty(c, name)
+
+Return the value of part with given `name` in the composition `c`.
+"""
+function getproperty(c::Composition{D,SYMS}, S::Symbol) where {D,SYMS}
+  if S == :parts
+    getfield(c, :parts)
+  else
+    for (i, SYM) in enumerate(SYMS)
+      S == SYM && (return c.parts[i])
+    end
+    @error "invalid part name '$S'"
+  end
+end
 
 # ------------
 # IO methods

diff --git a/test/compositions.jl b/test/compositions.jl
@@ -14,4 +14,10 @@
   @test names(-c) == names(c)
   @test names(c - c) == names(c)
   @test names(2c) == names(c)
+
+  # get part by name
+  c = Composition(a=3,b=2,c=1)
+  @test c.a == 3
+  @test c.b == 2
+  @test c.c == 1
 end