Support setproperty! (#23)

invenia · Aug 5, 2020 · e2b9867 · e2b9867
1 parent 60bbb4a
commit e2b9867
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/src/KeyedFrames.jl b/src/KeyedFrames.jl
@@ -101,6 +101,7 @@ const ColumnIndex = Union{Real, Symbol}
 frame(kf::KeyedFrame) = getfield(kf, :frame)
 Base.keys(kf::KeyedFrame) = getfield(kf, :key)
 Base.setindex!(kf::KeyedFrame, value, ind...) = setindex!(frame(kf), value, ind...)
+Base.setproperty!(kf::KeyedFrame, field::Symbol, value) = setproperty!(frame(kf), field, value)
 
 # I don't want to have to write the same function body several times, so...
 function _kf_getindex(kf::KeyedFrame, index...)

diff --git a/test/runtests.jl b/test/runtests.jl
@@ -157,6 +157,12 @@ using Test
         @test isequal(cp, KeyedFrame(DataFrame(;a=[10, 2, 3, 4, 5],d=[10, 5, 6, 7, 8]), :a))
     end
 
+    @testset "setproperty!" begin
+        cp = deepcopy(kf1)
+        cp.b = 10:-1:1
+        @test isequal(cp, KeyedFrame(DataFrame(; a=1:10, b=10:-1:1, c=3:12), [:a, :b]))
+    end
+
     @testset "first/last" begin
         # Don't assume that n will always equal 6
         @test first(kf1) isa KeyedFrame