Fix bug in pop! and add tests for ClassifiedCollections

JuliaCollections · Mar 24, 2016 · c63c56b · c63c56b
1 parent f1eea87
commit c63c56b
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/src/classifiedcollections.jl b/src/classifiedcollections.jl
@@ -49,4 +49,4 @@ function push!{K, C}(cc::ClassifiedCollections{K, C}, key::K, e)
     push!(c, e)
 end
 
-pop!{K}(cc::Accumulator{K}, key::K) = pop!(cc.map, key)
+pop!{K}(cc::ClassifiedCollections{K}, key::K) = pop!(cc.map, key)
diff --git a/test/test_classifiedcollections.jl b/test/test_classifiedcollections.jl
@@ -20,6 +20,12 @@ push!(c, "high", 5)
 @test c["low"] == [1, 2, 3]
 @test c["high"] == [4, 5]
 
+@test length(c) == 2
+@test collect(keys(c)) == ASCIIString["high","low"]
+
+pop!(c,"low")
+@test !haskey(c,"low")
+
 # classified sets
 
 c = classified_sets(ASCIIString, Int)