Add get(mmcif, key, default) method for MMCIFDict. (#24)

BioJulia · Sep 27, 2020 · e25b969 · e25b969
1 parent e58f6c1
commit e25b969
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/src/mmcif.jl b/src/mmcif.jl
@@ -72,6 +72,7 @@ end
 Base.keys(mmcif_dict::MMCIFDict) = keys(mmcif_dict.dict)
 Base.values(mmcif_dict::MMCIFDict) = values(mmcif_dict.dict)
 Base.haskey(mmcif_dict::MMCIFDict, key) = haskey(mmcif_dict.dict, key)
+Base.get(mmcif_dict::MMCIFDict, key, default) = get(mmcif_dict.dict, key, default)
 
 function Base.show(io::IO, mmcif_dict::MMCIFDict)
     print(io, "mmCIF dictionary with $(length(keys(mmcif_dict))) fields")

diff --git a/test/runtests.jl b/test/runtests.jl
@@ -1594,6 +1594,9 @@ end
     @test length(values(dic)) == 610
     @test haskey(dic, "_cell.entry_id")
     @test !haskey(dic, "nokey")
+    @test get(dic, "_cell.entry_id", ["default"]) == ["1AKE"]
+    @test get(dic, "nokey", ["default"]) == ["default"]
+    @test ismissing(get(dic, "nokey", missing))
     show(devnull, dic)
 
     multiline_str = """